Home
last modified time | relevance | path

Searched refs:AddBuiltin (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dregister.cc37 AddBuiltin(BuiltinOperator_ABS, Register_ABS(), /* min_version = */ 1, in BuiltinOpResolver()
39 AddBuiltin(BuiltinOperator_HARD_SWISH, Register_HARD_SWISH()); in BuiltinOpResolver()
40 AddBuiltin(BuiltinOperator_RELU, Register_RELU(), /* min_version = */ 1, in BuiltinOpResolver()
42 AddBuiltin(BuiltinOperator_RELU_N1_TO_1, Register_RELU_N1_TO_1()); in BuiltinOpResolver()
43 AddBuiltin(BuiltinOperator_RELU6, Register_RELU6(), /* min_version = */ 1, in BuiltinOpResolver()
45 AddBuiltin(BuiltinOperator_TANH, Register_TANH(), /* min_version = */ 1, in BuiltinOpResolver()
47 AddBuiltin(BuiltinOperator_LOGISTIC, Register_LOGISTIC(), in BuiltinOpResolver()
50 AddBuiltin(BuiltinOperator_AVERAGE_POOL_2D, Register_AVERAGE_POOL_2D(), in BuiltinOpResolver()
53 AddBuiltin(BuiltinOperator_MAX_POOL_2D, Register_MAX_POOL_2D(), in BuiltinOpResolver()
56 AddBuiltin(BuiltinOperator_L2_POOL_2D, Register_L2_POOL_2D()); in BuiltinOpResolver()
[all …]
Dregister_ref.cc197 AddBuiltin(BuiltinOperator_ABS, Register_ABS(), /* min_version = */ 1, in BuiltinRefOpResolver()
199 AddBuiltin(BuiltinOperator_HARD_SWISH, Register_HARD_SWISH_REF()); in BuiltinRefOpResolver()
200 AddBuiltin(BuiltinOperator_RELU, Register_RELU(), /* min_version = */ 1, in BuiltinRefOpResolver()
202 AddBuiltin(BuiltinOperator_RELU_N1_TO_1, Register_RELU_N1_TO_1()); in BuiltinRefOpResolver()
203 AddBuiltin(BuiltinOperator_RELU6, Register_RELU6(), /* min_version = */ 1, in BuiltinRefOpResolver()
205 AddBuiltin(BuiltinOperator_TANH, Register_TANH_REF(), /* min_version = */ 1, in BuiltinRefOpResolver()
207 AddBuiltin(BuiltinOperator_LOGISTIC, Register_LOGISTIC_REF(), in BuiltinRefOpResolver()
210 AddBuiltin(BuiltinOperator_AVERAGE_POOL_2D, Register_AVERAGE_POOL_REF(), in BuiltinRefOpResolver()
213 AddBuiltin(BuiltinOperator_MAX_POOL_2D, Register_MAX_POOL_REF(), in BuiltinRefOpResolver()
216 AddBuiltin(BuiltinOperator_L2_POOL_2D, Register_L2_POOL_REF()); in BuiltinRefOpResolver()
[all …]
/external/tensorflow/tensorflow/lite/micro/
Dmicro_mutable_op_resolver.h116 return AddBuiltin(BuiltinOperator_ABS, tflite::ops::micro::Register_ABS(), in AddAbs()
121 return AddBuiltin(BuiltinOperator_ADD, tflite::ops::micro::Register_ADD(), in AddAdd()
126 return AddBuiltin(BuiltinOperator_ARG_MAX, in AddArgMax()
131 return AddBuiltin(BuiltinOperator_ARG_MIN, in AddArgMin()
136 return AddBuiltin(BuiltinOperator_AVERAGE_POOL_2D, in AddAveragePool2D()
142 return AddBuiltin(BuiltinOperator_BATCH_TO_SPACE_ND, in AddBatchToSpaceND()
147 return AddBuiltin(BuiltinOperator_CAST, Register_CAST(), ParseCast); in AddCast()
151 return AddBuiltin(BuiltinOperator_CEIL, tflite::ops::micro::Register_CEIL(), in AddCeil()
161 return AddBuiltin(BuiltinOperator_CONCATENATION, in AddConcatenation()
167 return AddBuiltin(BuiltinOperator_CONV_2D, Register_CONV_2D(), ParseConv2D); in AddConv2D()
[all …]
/external/libtextclassifier/native/utils/
Dtflite-model-executor.cc103 resolver->AddBuiltin(tflite::BuiltinOperator_ADD, in RegisterSelectedOps()
107 resolver->AddBuiltin(tflite::BuiltinOperator_CONCATENATION, in RegisterSelectedOps()
111 resolver->AddBuiltin(tflite::BuiltinOperator_CONV_2D, in RegisterSelectedOps()
115 resolver->AddBuiltin(::tflite::BuiltinOperator_EQUAL, in RegisterSelectedOps()
118 resolver->AddBuiltin(tflite::BuiltinOperator_FULLY_CONNECTED, in RegisterSelectedOps()
122 resolver->AddBuiltin(::tflite::BuiltinOperator_GREATER_EQUAL, in RegisterSelectedOps()
124 resolver->AddBuiltin(tflite::BuiltinOperator_L2_NORMALIZATION, in RegisterSelectedOps()
128 resolver->AddBuiltin(tflite::BuiltinOperator_MUL, in RegisterSelectedOps()
130 resolver->AddBuiltin(tflite::BuiltinOperator_RESHAPE, in RegisterSelectedOps()
132 resolver->AddBuiltin(::tflite::BuiltinOperator_REDUCE_MAX, in RegisterSelectedOps()
[all …]
/external/tensorflow/tensorflow/lite/
Dmutable_op_resolver_test.cc69 resolver.AddBuiltin(BuiltinOperator_ADD, GetDummyRegistration()); in TEST()
81 resolver.AddBuiltin(BuiltinOperator_ADD, GetDummyRegistration()); in TEST()
91 resolver.AddBuiltin(BuiltinOperator_ADD, GetDummyRegistration(), 2); in TEST()
110 resolver.AddBuiltin(BuiltinOperator_ADD, GetDummyRegistration(), 2, 3); in TEST()
128 resolver.AddBuiltin(BuiltinOperator_ADD, GetDummyRegistration(), 2, 3); in TEST()
171 resolver1.AddBuiltin(BuiltinOperator_ADD, reg); in TEST()
204 resolver1.AddBuiltin(BuiltinOperator_ADD, GetDummyRegistration()); in TEST()
205 resolver1.AddBuiltin(BuiltinOperator_MUL, GetDummy2Registration()); in TEST()
208 resolver2.AddBuiltin(BuiltinOperator_SUB, GetDummyRegistration()); in TEST()
209 resolver2.AddBuiltin(BuiltinOperator_ADD, GetDummy2Registration()); in TEST()
Dmutable_op_resolver.cc39 void MutableOpResolver::AddBuiltin(tflite::BuiltinOperator op, in AddBuiltin() function in tflite::MutableOpResolver
50 void MutableOpResolver::AddBuiltin(tflite::BuiltinOperator op, in AddBuiltin() function in tflite::MutableOpResolver
54 AddBuiltin(op, registration, version); in AddBuiltin()
Dmutable_op_resolver.h67 void AddBuiltin(tflite::BuiltinOperator op,
69 void AddBuiltin(tflite::BuiltinOperator op,
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dlogging_op_resolver_test.cc60 base_resolver.AddBuiltin(BuiltinOperator_CONV_2D, &conv_registration); in TEST()
66 base_resolver.AddBuiltin(BuiltinOperator_ADD, &add_registration); in TEST()
94 base_resolver.AddBuiltin(BuiltinOperator_CONV_2D, &conv_registration); in TEST()
100 base_resolver.AddBuiltin(BuiltinOperator_ADD, &add_registration); in TEST()
121 base_resolver.AddBuiltin(BuiltinOperator_CONV_2D, &conv_registration); in TEST()
127 base_resolver.AddBuiltin(BuiltinOperator_ADD, &add_registration); in TEST()
/external/libtextclassifier/native/annotator/pod_ner/
Dpod-ner-impl.cc118 mutable_resolver->AddBuiltin(::tflite::BuiltinOperator_SHAPE, in CreateInterpreter()
120 mutable_resolver->AddBuiltin(::tflite::BuiltinOperator_RANGE, in CreateInterpreter()
122 mutable_resolver->AddBuiltin( in CreateInterpreter()
125 mutable_resolver->AddBuiltin( in CreateInterpreter()
/external/tensorflow/tensorflow/lite/c/
Dc_api_experimental.cc38 options->mutable_op_resolver.AddBuiltin( in TfLiteInterpreterOptionsAddBuiltinOp()
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dops_version.md160 void AddBuiltin(tflite::BuiltinOperator op, TfLiteRegistration* registration,
171 AddBuiltin(BuiltinOperator_DEPTHWISE_CONV_2D, Register_DEPTHWISE_CONV_2D());
177 AddBuiltin(BuiltinOperator_DEPTHWISE_CONV_2D, Register_DEPTHWISE_CONV_2D(),
Dops_custom.md256 virtual void AddBuiltin(tflite::BuiltinOperator op, TfLiteRegistration* registration) = 0;
289 implementations of builtins by using the `AddBuiltin`.
/external/tensorflow/tensorflow/lite/tools/
Dverifier_test.cc76 resolver_.AddBuiltin(iter, &fake_op_); in TfLiteFlatbufferModelBuilder()