Skip to content

Commit 970dc9c

Browse files
committed
Bugfix after latest sync: use micro_mutable_ops_resolver
Signed-off-by: Vikram <[email protected]>
1 parent de0df78 commit 970dc9c

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

examples/hello_world/main/main_functions.cc

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ limitations under the License.
1414
==============================================================================*/
1515

1616

17-
#include "tensorflow/lite/micro/all_ops_resolver.h"
17+
#include "tensorflow/lite/micro/micro_mutable_op_resolver.h"
1818
#include "tensorflow/lite/micro/micro_interpreter.h"
1919
#include "tensorflow/lite/micro/system_setup.h"
2020
#include "tensorflow/lite/schema/schema_generated.h"
@@ -47,9 +47,11 @@ void setup() {
4747
return;
4848
}
4949

50-
// This pulls in all the operation implementations we need.
51-
// NOLINTNEXTLINE(runtime-global-variables)
52-
static tflite::AllOpsResolver resolver;
50+
// Pull in only the operation implementations we need.
51+
static tflite::MicroMutableOpResolver<1> resolver;
52+
if (resolver.AddFullyConnected() != kTfLiteOk) {
53+
return;
54+
}
5355

5456
// Build an interpreter to run the model with.
5557
static tflite::MicroInterpreter static_interpreter(

0 commit comments

Comments
 (0)