Home
last modified time | relevance | path

Searched refs:SetCustomOp (Results 1 – 14 of 14) sorted by relevance

/external/libtextclassifier/utils/tflite/
Ddist_diversification_test.cc62 SetCustomOp("DistanceDiversification", {}, in DistanceDiversificationOpModel()
Dtoken_encoder_test.cc85 SetCustomOp("TokenEncoder", {}, tflite::ops::custom::Register_TOKEN_ENCODER); in TokenEncoderOpModel()
Dtext_encoder_test.cc105 SetCustomOp("TextEncoder", builder.GetBuffer(), in TextEncoderOpModel()
/external/tensorflow/tensorflow/lite/models/smartreply/ops/
Dnormalize_test.cc41 SetCustomOp("Normalize", {}, Register_NORMALIZE); in NormalizeOpModel()
Dextract_feature_test.cc42 SetCustomOp("ExtractFeatures", {}, Register_EXTRACT_FEATURES); in ExtractFeatureOpModel()
Dpredict_test.cc51 SetCustomOp("Predict", predict_option, Register_PREDICT); in PredictOpModel()
/external/tensorflow/tensorflow/lite/kernels/
Dmfcc_test.cc54 SetCustomOp("Mfcc", fbb.GetBuffer(), Register_MFCC); in BaseMfccOpModel()
Daudio_spectrogram_test.cc53 SetCustomOp("AudioSpectrogram", fbb.GetBuffer(), in BaseAudioSpectrogramOpModel()
Ddetection_postprocess_test.cc68 SetCustomOp("TFLite_Detection_PostProcess", fbb.GetBuffer(), in BaseDetectionPostprocessOpModel()
279 SetCustomOp("TFLite_Detection_PostProcess", fbb.GetBuffer(), in DetectionPostprocessOpModelwithRegularNMS()
Dtest_util.cc83 void SingleOpModel::SetCustomOp( in SetCustomOp() function in tflite::SingleOpModel
Dtest_util.h246 void SetCustomOp(const string& name,
/external/tensorflow/tensorflow/lite/profiling/
Dprofile_summarizer_test.cc94 SetCustomOp(kOpName, {}, registration); in Init()
/external/tensorflow/tensorflow/lite/experimental/microfrontend/
Daudio_microfrontend_test.cc82 SetCustomOp("MICRO_FRONTEND", fbb.GetBuffer(), in MicroFrontendOpModel()
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dctc_beam_search_decoder_test.cc62 SetCustomOp("CTCBeamSearchDecoder", fbb.GetBuffer(), in CTCBeamSearchDecoderOpModel()