Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/python/
Dtest_util_test.py21 from tensorflow.lite.python import test_util as tflite_test_util unknown
32 op_set = tflite_test_util.get_ops_list(gfile.GFile(model_path, 'rb').read())
38 op_set = tflite_test_util.get_ops_list(gfile.GFile(model_path, 'rb').read())
Dlite_flex_test.py28 from tensorflow.lite.python import test_util as tflite_test_util unknown
92 self.assertIn('FlexL2Loss', tflite_test_util.get_ops_list(tflite_model))
210 self.assertIn('FlexCustomAdd4', tflite_test_util.get_ops_list(tflite_model))
229 self.assertIn('FlexDouble', tflite_test_util.get_ops_list(tflite_model))
Dtflite_convert_test.py28 from tensorflow.lite.python import test_util as tflite_test_util unknown
71 op_set = tflite_test_util.get_ops_list(content)
75 output_shapes = tflite_test_util.get_output_shapes(content)
Dlite_v2_test.py43 from tensorflow.lite.python import test_util as tflite_test_util unknown
2890 self.assertIn('Double', tflite_test_util.get_ops_list(tflite_model))
2920 self.assertIn('Double', 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)