/external/armnn/python/pyarmnn/test/ |
D | test_onnx_parser.py | 12 def parser(shared_data_folder): argument 21 path_to_model = os.path.join(shared_data_folder, 'mock_model.onnx') 63 def test_onnx_filenotfound_exception(shared_data_folder): argument 67 path_to_model = os.path.join(shared_data_folder, 'some_unknown_model.onnx') 79 def test_onnx_parser_end_to_end(shared_data_folder): argument 82 … network = parser.CreateNetworkFromBinaryFile(os.path.join(shared_data_folder, 'mock_model.onnx')) 86 …input_tensor_data = np.load(os.path.join(shared_data_folder, 'onnx_parser/input_onnx.npy')).astype… 108 golden_output = np.load(os.path.join(shared_data_folder, 'onnx_parser/golden_output_onnx.npy'))
|
D | test_deserializer.py | 11 def parser(shared_data_folder): argument 16 parser.CreateNetworkFromBinary(os.path.join(shared_data_folder, 'mock_model.armnn')) 65 def test_deserializer_filenotfound_exception(shared_data_folder): argument 69 … parser.CreateNetworkFromBinary(os.path.join(shared_data_folder, 'some_unknown_network.armnn')) 76 def test_deserializer_end_to_end(shared_data_folder): argument 79 network = parser.CreateNetworkFromBinary(os.path.join(shared_data_folder, "mock_model.armnn")) 100 input_tensor_data = np.load(os.path.join(shared_data_folder, 'deserializer/input_lite.npy')) 117 …expected_outputs = np.load(os.path.join(shared_data_folder, 'deserializer/golden_output_lite.npy'))
|
D | test_caffe_parser.py | 11 def parser(shared_data_folder): argument 20 path_to_model = os.path.join(shared_data_folder, 'mock_model.caffemodel') 66 def test_filenotfound_exception(shared_data_folder): argument 70 path_to_model = os.path.join(shared_data_folder, 'some_unknown_network.caffemodel') 86 def test_caffe_parser_end_to_end(shared_data_folder): argument 94 …network = parser.CreateNetworkFromBinaryFile(os.path.join(shared_data_folder, 'mock_model.caffemod… 114 …input_tensor_data = np.load(os.path.join(shared_data_folder, 'caffe_parser/input_caffe.npy')).asty… 128 …expected_output = np.load(os.path.join(shared_data_folder, 'caffe_parser/golden_output_caffe.npy'))
|
D | test_tf_parser.py | 11 def parser(shared_data_folder): argument 20 path_to_model = os.path.join(shared_data_folder, 'mock_model.pb') 68 def test_tf_filenotfound_exception(shared_data_folder): argument 72 path_to_model = os.path.join(shared_data_folder, 'some_unknown_model.pb') 90 def test_tf_parser_end_to_end(shared_data_folder): argument 96 network = parser.CreateNetworkFromBinaryFile(os.path.join(shared_data_folder, 'mock_model.pb'), 102 …input_tensor_data = np.load(os.path.join(shared_data_folder, 'tf_parser/input_tf.npy')).astype(np.… 130 golden_output = np.load(os.path.join(shared_data_folder, 'tf_parser/golden_output_tf.npy'))
|
D | test_tflite_parser.py | 17 def parser(shared_data_folder): argument 22 parser.CreateNetworkFromBinaryFile(os.path.join(shared_data_folder, 'mock_model.tflite')) 51 def test_tflite_parser_with_optional_options_out_of_scope(shared_data_folder): argument 53 …network = parser.CreateNetworkFromBinaryFile(os.path.join(shared_data_folder, "mock_model.tflite")) 132 def test_tflite_filenotfound_exception(shared_data_folder): argument 136 …parser.CreateNetworkFromBinaryFile(os.path.join(shared_data_folder, 'some_unknown_network.tflite')) 143 def test_tflite_parser_end_to_end(shared_data_folder): argument 146 …network = parser.CreateNetworkFromBinaryFile(os.path.join(shared_data_folder, "mock_model.tflite")) 168 input_tensor_data = np.load(os.path.join(shared_data_folder, 'tflite_parser/input_lite.npy')) 186 …expected_outputs = np.load(os.path.join(shared_data_folder, 'tflite_parser/golden_output_lite.npy'…
|
D | test_tensor_conversion.py | 11 def get_tensor_info_input(shared_data_folder): argument 16 parser.CreateNetworkFromBinaryFile(os.path.join(shared_data_folder, 'mock_model.tflite')) 25 def get_tensor_info_output(shared_data_folder): argument 30 parser.CreateNetworkFromBinaryFile(os.path.join(shared_data_folder, 'mock_model.tflite'))
|
D | test_runtime.py | 12 def random_runtime(shared_data_folder): argument 14 …network = parser.CreateNetworkFromBinaryFile(os.path.join(shared_data_folder, 'mock_model.tflite')) 53 def mock_model_runtime(shared_data_folder): argument 55 …network = parser.CreateNetworkFromBinaryFile(os.path.join(shared_data_folder, 'mock_model.tflite')) 60 input_tensor_data = np.load(os.path.join(shared_data_folder, 'tflite_parser/input_lite.npy'))
|
D | test_profiling_utilities.py | 19 def mock_profiler(shared_data_folder): argument 20 path_to_file = os.path.join(shared_data_folder, 'mock_profile_out.json')
|
D | test_supported_backends.py | 10 def get_supported_backends_setup(shared_data_folder): argument
|
D | test_network.py | 38 def get_runtime(shared_data_folder, network_file): argument 41 network = parser.CreateNetworkFromBinaryFile(os.path.join(shared_data_folder, network_file))
|
/external/armnn/python/pyarmnn/ |
D | conftest.py | 22 def shared_data_folder(request): function
|