Home
last modified time | relevance | path

Searched refs:micro_op_resolver (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/
Dmicro_mutable_op_resolver_test.cc54 MicroOpResolver<7> micro_op_resolver; in TF_LITE_MICRO_TEST() local
55 micro_op_resolver.AddBuiltin(BuiltinOperator_CONV_2D, &r, 0, 2); in TF_LITE_MICRO_TEST()
56 micro_op_resolver.AddCustom("mock_custom", &r, 0, 3); in TF_LITE_MICRO_TEST()
57 OpResolver* resolver = &micro_op_resolver; in TF_LITE_MICRO_TEST()
66 TF_LITE_MICRO_EXPECT_EQ(7, micro_op_resolver.GetRegistrationLength()); in TF_LITE_MICRO_TEST()
96 MicroOpResolver<4> micro_op_resolver; in TF_LITE_MICRO_TEST() local
99 micro_op_resolver.AddBuiltin(BuiltinOperator_CONV_2D, &r, 0, 2); in TF_LITE_MICRO_TEST()
100 micro_op_resolver.AddCustom("mock_custom", &r, 0, 3); in TF_LITE_MICRO_TEST()
101 OpResolver* resolver = &micro_op_resolver; in TF_LITE_MICRO_TEST()
103 TF_LITE_MICRO_EXPECT_EQ(4, micro_op_resolver.GetRegistrationLength()); in TF_LITE_MICRO_TEST()
/external/tensorflow/tensorflow/lite/micro/examples/person_detection_experimental/
Dmain_functions.cc75 static tflite::MicroOpResolver<12> micro_op_resolver; in setup() local
76 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_DEPTHWISE_CONV_2D, in setup()
79 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_CONV_2D, in setup()
81 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_AVERAGE_POOL_2D, in setup()
84 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_RESHAPE, in setup()
86 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_SOFTMAX, in setup()
92 model, micro_op_resolver, tensor_arena, kTensorArenaSize, error_reporter); in setup()
Dperson_detection_test.cc55 tflite::MicroOpResolver<11> micro_op_resolver; in TF_LITE_MICRO_TEST() local
56 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_DEPTHWISE_CONV_2D, in TF_LITE_MICRO_TEST()
59 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_CONV_2D, in TF_LITE_MICRO_TEST()
61 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_AVERAGE_POOL_2D, in TF_LITE_MICRO_TEST()
64 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_RESHAPE, in TF_LITE_MICRO_TEST()
66 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_SOFTMAX, in TF_LITE_MICRO_TEST()
70 tflite::MicroInterpreter interpreter(model, micro_op_resolver, tensor_arena, in TF_LITE_MICRO_TEST()
/external/tensorflow/tensorflow/lite/micro/examples/magic_wand/
Dmain_functions.cc70 static tflite::MicroOpResolver<5> micro_op_resolver; // NOLINT in setup() local
71 micro_op_resolver.AddBuiltin( in setup()
74 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_MAX_POOL_2D, in setup()
76 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_CONV_2D, in setup()
78 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_FULLY_CONNECTED, in setup()
80 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_SOFTMAX, in setup()
85 model, micro_op_resolver, tensor_arena, kTensorArenaSize, error_reporter); in setup()
Dmagic_wand_test.cc49 static tflite::MicroOpResolver<5> micro_op_resolver; // NOLINT in TF_LITE_MICRO_TEST() local
50 micro_op_resolver.AddBuiltin( in TF_LITE_MICRO_TEST()
53 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_MAX_POOL_2D, in TF_LITE_MICRO_TEST()
55 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_CONV_2D, in TF_LITE_MICRO_TEST()
57 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_FULLY_CONNECTED, in TF_LITE_MICRO_TEST()
59 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_SOFTMAX, in TF_LITE_MICRO_TEST()
68 tflite::MicroInterpreter interpreter(model, micro_op_resolver, tensor_arena, in TF_LITE_MICRO_TEST()
/external/tensorflow/tensorflow/lite/micro/examples/person_detection/
Dmain_functions.cc68 static tflite::MicroOpResolver<3> micro_op_resolver; in setup() local
69 micro_op_resolver.AddBuiltin( in setup()
72 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_CONV_2D, in setup()
74 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_AVERAGE_POOL_2D, in setup()
79 model, micro_op_resolver, tensor_arena, kTensorArenaSize, error_reporter); in setup()
Dperson_detection_test.cc57 tflite::MicroOpResolver<3> micro_op_resolver; in TF_LITE_MICRO_TEST() local
58 micro_op_resolver.AddBuiltin( in TF_LITE_MICRO_TEST()
61 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_CONV_2D, in TF_LITE_MICRO_TEST()
63 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_AVERAGE_POOL_2D, in TF_LITE_MICRO_TEST()
67 tflite::MicroInterpreter interpreter(model, micro_op_resolver, tensor_arena, in TF_LITE_MICRO_TEST()
/external/tensorflow/tensorflow/lite/micro/examples/micro_speech/
Dmain_functions.cc77 static tflite::MicroOpResolver<3> micro_op_resolver; in setup() local
78 micro_op_resolver.AddBuiltin( in setup()
81 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_FULLY_CONNECTED, in setup()
83 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_SOFTMAX, in setup()
88 model, micro_op_resolver, tensor_arena, kTensorArenaSize, error_reporter); in setup()
Dmicro_speech_test.cc52 tflite::MicroOpResolver<3> micro_op_resolver; in TF_LITE_MICRO_TEST() local
53 micro_op_resolver.AddBuiltin( in TF_LITE_MICRO_TEST()
56 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_FULLY_CONNECTED, in TF_LITE_MICRO_TEST()
58 micro_op_resolver.AddBuiltin(tflite::BuiltinOperator_SOFTMAX, in TF_LITE_MICRO_TEST()
66 tflite::MicroInterpreter interpreter(model, micro_op_resolver, tensor_arena, in TF_LITE_MICRO_TEST()