| /external/tensorflow/tensorflow/lite/kernels/ |
| D | random_uniform_test.cc | 62 SetCustomOp("RandomUniform", {}, ops::custom::Register_RANDOM_UNIFORM); in RandomUniformOpModel()
90 SetCustomOp("RandomUniformInt", {}, in RandomUniformIntOpModel()
|
| D | random_standard_normal_test.cc | 52 SetCustomOp("RandomStandardNormal", {}, in RandomStandardNormalOpModel()
|
| D | audio_spectrogram_test.cc | 50 SetCustomOp("AudioSpectrogram", fbb.GetBuffer(), in BaseAudioSpectrogramOpModel()
|
| D | mfcc_test.cc | 52 SetCustomOp("Mfcc", fbb.GetBuffer(), Register_MFCC); in BaseMfccOpModel()
|
| D | numeric_verify_test.cc | 64 SetCustomOp("NUMERIC_VERIFY", fbb.GetBuffer(), in NumericVerifyOpModel()
|
| D | multinomial_test.cc | 62 SetCustomOp("Multinomial", {}, ops::custom::Register_MULTINOMIAL); in MultinomialOpModel()
|
| D | detection_postprocess_test.cc | 69 SetCustomOp("TFLite_Detection_PostProcess", fbb.GetBuffer(), in BaseDetectionPostprocessOpModel()
344 SetCustomOp("TFLite_Detection_PostProcess", fbb.GetBuffer(), in DetectionPostprocessOpModelwithRegularNMS()
|
| D | test_util.cc | 147 void SingleOpModel::SetCustomOp( in SetCustomOp() function in tflite::SingleOpModel
|
| D | test_util.h | 478 void SetCustomOp(const string& name,
|
| /external/tensorflow/tensorflow/lite/kernels/perception/ |
| D | dense_image_warp_test.cc | 41 SetCustomOp("DenseImageWarp", custom_option, RegisterDenseImageWarp); in DenseImageWarpOpModel()
|
| D | max_unpooling_2d_test.cc | 48 SetCustomOp("MaxUnpooling2D", custom_option, RegisterMaxUnpooling2D); in MaxUnpoolingOpModel()
|
| D | max_pool_with_argmax_test.cc | 47 SetCustomOp("MaxPoolWithArgmax", custom_option, RegisterMaxPoolWithArgmax); in MaxpoolingWithArgMaxOpModel()
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | unidirectional_sequence_gru_test.cc | 49 SetCustomOp("UNIDIRECTIONAL_SEQUENCE_GRU", {}, in GRUOpModel()
|
| D | ctc_beam_search_decoder_test.cc | 62 SetCustomOp("CTCBeamSearchDecoder", fbb.GetBuffer(), in CTCBeamSearchDecoderOpModel()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| D | ragged_range_tflite_test.cc | 56 SetCustomOp("RaggedRange", {}, ops::custom::Register_RAGGED_RANGE); in RaggedRangeOpModel()
|
| D | ragged_tensor_to_tensor_tflite_test.cc | 74 SetCustomOp("RaggedTensorToTensor", fbb.GetBuffer(), in RaggedTensorToTensorOpModel()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| D | whitespace_tokenizer_test.cc | 54 SetCustomOp("WhitespaceTokenizer", {}, Register_tftext_WhitespaceTokenizer); in WhitespaceTokenizerModel()
|
| D | ngrams_test.cc | 144 SetCustomOp("tftext:Ngrams", fbb.GetBuffer(), Register_tftext_Ngrams); in BuildCustomOp()
|
| /external/tensorflow/tensorflow/lite/experimental/microfrontend/ |
| D | audio_microfrontend_test.cc | 81 SetCustomOp("MICRO_FRONTEND", fbb.GetBuffer(), in MicroFrontendOpModel()
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | list_flex_ops_test.cc | 78 SetCustomOp(op_name, custom_options, Register_TEST); in FlexOpModel()
|
| /external/tensorflow/tensorflow/lite/kernels/hashtable/ |
| D | hashtable_ops_test.cc | 649 SetCustomOp("HASHTABLE", fbb.GetBuffer(), in HashtableOpModel()
812 SetCustomOp("HASHTABLE_FIND", {}, in HashtableFindOpModel()
888 SetCustomOp("HASHTABLE_IMPORT", {}, in HashtableImportOpModel()
965 SetCustomOp("HASHTABLE_SIZE", {}, in HashtableSizeOpModel()
|
| /external/tensorflow/tensorflow/lite/profiling/ |
| D | profile_summarizer_test.cc | 98 SetCustomOp(kOpName, {}, registration); in Init()
|
| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | parse_example_test.cc | 208 SetCustomOp("ParseExample", buffer, Register_PARSE_EXAMPLE); in ParseExampleOpModel()
|