| /external/tensorflow/tensorflow/lite/kernels/hashtable/ |
| D | hashtable_ops.cc | 24 resolver->AddCustom("HashTableV2", tflite::ops::custom::Register_HASHTABLE()); in AddHashtableOps()
25 resolver->AddCustom("LookupTableFindV2", in AddHashtableOps()
27 resolver->AddCustom("LookupTableImportV2", in AddHashtableOps()
29 resolver->AddCustom("LookupTableSizeV2", in AddHashtableOps()
|
| /external/tensorflow/tensorflow/lite/kernels/perception/ |
| D | perception_ops.cc | 23 resolver->AddCustom("DenseImageWarp", in AddPerceptionOps()
25 resolver->AddCustom("MaxPoolWithArgmax", in AddPerceptionOps()
27 resolver->AddCustom("MaxUnpooling2D", in AddPerceptionOps()
|
| /external/tensorflow/tensorflow/lite/micro/ |
| D | micro_mutable_op_resolver_test.cc | 76 micro_op_resolver.AddCustom("mock_custom", &r)); in TF_LITE_MICRO_TEST()
80 micro_op_resolver.AddCustom("mock_custom", &r)); in TF_LITE_MICRO_TEST()
118 micro_op_resolver.AddCustom("mock_custom_0", &r)); in TF_LITE_MICRO_TEST()
127 micro_op_resolver.AddCustom("mock_custom_1", &r)); in TF_LITE_MICRO_TEST()
|
| D | micro_mutable_op_resolver.h | 82 TfLiteStatus AddCustom(const char* name, TfLiteRegistration* registration) { in AddCustom() function
156 return AddCustom("CIRCULAR_BUFFER", in AddCircularBuffer()
187 return AddCustom("TFLite_Detection_PostProcess", in AddDetectionPostprocess()
199 return AddCustom(tflite::GetString_ETHOSU(), registration); in AddEthosU()
|
| D | test_helpers.cc | 830 op_resolver.AddCustom("mock_custom", MockCustom::GetMutableRegistration()); in GetOpResolver()
831 op_resolver.AddCustom("simple_stateful_op", in GetOpResolver()
833 op_resolver.AddCustom("multiple_inputs_op", in GetOpResolver()
|
| /external/libtextclassifier/native/utils/ |
| D | tflite-model-executor.cc | 280 resolver->AddCustom("DistanceDiversification", in BuildOpResolver()
282 resolver->AddCustom("TextEncoder", in BuildOpResolver()
284 resolver->AddCustom("TokenEncoder", in BuildOpResolver()
286 resolver->AddCustom( in BuildOpResolver()
289 resolver->AddCustom("RaggedRange", in BuildOpResolver()
291 resolver->AddCustom( in BuildOpResolver()
294 resolver->AddCustom( in BuildOpResolver()
297 resolver->AddCustom( in BuildOpResolver()
300 resolver->AddCustom("RandomUniform", in BuildOpResolver()
|
| /external/tensorflow/tensorflow/lite/ |
| D | mutable_op_resolver.cc | 58 void MutableOpResolver::AddCustom(const char* name, in AddCustom() function in tflite::MutableOpResolver
69 void MutableOpResolver::AddCustom(const char* name, in AddCustom() function in tflite::MutableOpResolver
73 AddCustom(name, registration, version); in AddCustom()
|
| D | mutable_op_resolver_test.cc | 141 resolver.AddCustom("AWESOME", GetDummyRegistration()); in TEST()
155 resolver.AddCustom(reg->custom_name, reg); in TEST()
187 resolver.AddCustom("AWESOME", GetDummyRegistration()); in TEST()
196 resolver.AddCustom("AWESOME", GetDummyRegistration()); in TEST()
|
| D | mutable_op_resolver.h | 72 void AddCustom(const char* name, const TfLiteRegistration* registration,
74 void AddCustom(const char* name, const TfLiteRegistration* registration,
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/ |
| D | py_tflite_registerer.cc | 29 resolver->AddCustom("TFSentencepieceTokenizeOp", in TFLite_SentencepieceTokenizerRegisterer()
31 resolver->AddCustom( in TFLite_SentencepieceTokenizerRegisterer()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| D | ngrams_op_resolver.cc | 26 resolver->AddCustom("tftext:Ngrams", Register_tftext_Ngrams()); in AddNgramsCustomOp()
|
| D | whitespace_tokenizer_op_resolver.cc | 26 resolver->AddCustom("tftext:WhitespaceTokenizer", in AddWhitespaceTokenizerCustomOp()
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | verifier_test.cc | 79 resolver_.AddCustom(iter.data(), &fake_op_); in TfLiteFlatbufferModelBuilder()
593 resolver.AddCustom("FakeOp", &fake_op); in TEST()
617 resolver.AddCustom("FakeOp", &fake_op); in TEST()
642 resolver.AddCustom("FakeOp", &fake_op); in TEST()
666 resolver.AddCustom("FakeOp", &fake_op); in TEST()
692 resolver.AddCustom("FakeOp", &fake_op); in TEST()
730 resolver.AddCustom("FakeOp", &fake_op); in TEST()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| D | py_tflite_registerer.cc | 29 resolver->AddCustom("RaggedTensorToTensor", in TFLite_RaggedTensorToTensorRegisterer()
|
| /external/tensorflow/tensorflow/lite/python/testdata/ |
| D | test_registerer.cc | 34 resolver->AddCustom("FakeOp", GetFakeRegistration()); in TF_TestRegisterer()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | c_api_experimental.cc | 56 options->mutable_op_resolver.AddCustom(name, registration, min_version, in TfLiteInterpreterOptionsAddCustomOp()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | register.cc | 321 AddCustom("NumericVerify", tflite::ops::custom::Register_NUMERIC_VERIFY()); in BuiltinOpResolver()
327 AddCustom("TFLite_Detection_PostProcess", in BuiltinOpResolver()
|
| D | register_ref.cc | 472 AddCustom("NumericVerify", in BuiltinRefOpResolver()
479 AddCustom("TFLite_Detection_PostProcess", in BuiltinRefOpResolver()
|
| D | test_util.cc | 201 resolver->AddCustom(reg.first.data(), reg.second()); in BuildInterpreter()
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | ops_custom.md | 257 virtual void AddCustom(const char* op, TfLiteRegistration* registration) = 0;
271 resolver.AddCustom("Sin", Register_SIN());
287 operations instead of a single operator. Just add as many `AddCustom` operators
296 TensorFlow Lite aware of your custom op by adding the appropriate `AddCustom`
|
| D | ops_version.md | 162 void AddCustom(const char* name, TfLiteRegistration* registration,
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | logging_op_resolver_test.cc | 151 base_resolver.AddCustom(custom_op_name.c_str(), &custom_registration); in TEST()
|
| /external/libtextclassifier/native/annotator/pod_ner/ |
| D | pod-ner-impl.cc | 128 mutable_resolver->AddCustom( in CreateInterpreter()
|
| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | parse_example.cc | 998 resolver->AddCustom("ParseExample", Register_PARSE_EXAMPLE()); in AddParseExampleOp()
999 resolver->AddCustom("ParseExampleV2", Register_PARSE_EXAMPLE_V2()); in AddParseExampleOp()
|