Searched refs:tflite_test_util (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/lite/python/ |
D | test_util_test.py | 17 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())
|
D | lite_flex_test.py | 25 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))
|
D | tflite_convert_test.py | 24 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)
|
D | lite_v2_test.py | 39 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/ |
D | mlir_convert.py | 22 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/ |
D | mlir_convert.py | 25 from tensorflow.lite.python import test_util as tflite_test_util unknown 112 ops_list = tflite_test_util.get_ops_list(tflite_model)
|