Home
last modified time | relevance | path

Searched refs:custom_op_registerers (Results 1 – 7 of 7) sorted by relevance

/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/
Dngrams_test.py119 model_content=model, custom_op_registerers=['AddNgramsCustomOp'])
139 model_content=model, custom_op_registerers=['AddNgramsCustomOp'])
158 model_content=model, custom_op_registerers=['AddNgramsCustomOp'])
189 model_content=model, custom_op_registerers=['AddNgramsCustomOp'])
219 model_content=model, custom_op_registerers=['AddNgramsCustomOp'])
Dwhitespace_tokenizer_test.py75 custom_op_registerers=['AddWhitespaceTokenizerCustomOp'])
101 custom_op_registerers=['AddWhitespaceTokenizerCustomOp'])
127 custom_op_registerers=['AddWhitespaceTokenizerCustomOp'])
/external/tflite-support/tensorflow_lite_support/custom_ops/python/
Dsentencepiece_tokenizer_test.py131 custom_op_registerers=["TFLite_SentencepieceTokenizerRegisterer"])
174 custom_op_registerers=["TFLite_SentencepieceTokenizerRegisterer"])
212 custom_op_registerers=["TFLite_SentencepieceTokenizerRegisterer"])
Dragged_tensor_to_tensor_test.py43 custom_op_registerers=["TFLite_RaggedTensorToTensorRegisterer"])
/external/tensorflow/tensorflow/lite/python/
Dinterpreter_test.py54 custom_op_registerers=['TF_TestRegisterer'])
62 custom_op_registerers=[test_registerer.TF_TestRegisterer])
73 custom_op_registerers=[bogus_name])
283 custom_op_registerers=['TF_TestRegisterer'])
Dinterpreter.py956 def __init__(self, custom_op_registerers=None, **kwargs): argument
969 self._custom_op_registerers = custom_op_registerers or []
Dlite_v2_test.py4131 model_content=tflite_model, custom_op_registerers=['TF_TestRegisterer'])
4162 custom_op_registerers=[test_registerer.TF_TestRegisterer])