Home
last modified time | relevance | path

Searched refs:tflite_test_util (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/python/
Dtest_util_test.py17 from tensorflow.lite.python import test_util as tflite_test_util unknown
28 op_set = tflite_test_util.get_ops_list(gfile.GFile(model_path, 'rb').read())
34 op_set = tflite_test_util.get_ops_list(gfile.GFile(model_path, 'rb').read())
Dlite_flex_test.py25 from tensorflow.lite.python import test_util as tflite_test_util unknown
90 self.assertIn('FlexL2Loss', tflite_test_util.get_ops_list(tflite_model))
251 self.assertIn('FlexCustomAdd4', tflite_test_util.get_ops_list(tflite_model))
270 self.assertIn('FlexDouble', tflite_test_util.get_ops_list(tflite_model))
349 self.assertIn('ADD', tflite_test_util.get_ops_list(tflite_model))
351 self.assertIn('FlexAddV2', tflite_test_util.get_ops_list(tflite_model))
379 self.assertIn('FlexL2Loss', tflite_test_util.get_ops_list(tflite_model))
425 tflite_test_util.get_ops_list(tflite_model))
Dtflite_convert_test.py24 from tensorflow.lite.python import test_util as tflite_test_util unknown
67 op_set = tflite_test_util.get_ops_list(content)
71 output_shapes = tflite_test_util.get_output_shapes(content)
Dlite_v2_test.py39 from tensorflow.lite.python import test_util as tflite_test_util unknown
4123 self.assertIn('Double', tflite_test_util.get_ops_list(tflite_model))
4157 self.assertIn('Double', tflite_test_util.get_ops_list(tflite_model))
/external/tensorflow/tensorflow/lite/testing/
Dmlir_convert.py22 from tensorflow.lite.python import test_util as tflite_test_util unknown
119 ops_list = tflite_test_util.get_ops_list(tflite_model)
/external/tensorflow/tensorflow/lite/experimental/mlir/testing/
Dmlir_convert.py25 from tensorflow.lite.python import test_util as tflite_test_util unknown
112 ops_list = tflite_test_util.get_ops_list(tflite_model)