diff options
author | Jan Eilers <jan.eilers@arm.com> | 2020-08-12 14:59:06 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2020-08-17 16:10:54 +0000 |
commit | 841aca155b35cc17ea9527599d2c364695e28166 (patch) | |
tree | 8861d6f275f5955495201f0e1b0677bb44604a17 /python/pyarmnn/test/test_tflite_parser.py | |
parent | 313c99f9a64c7fc51dc70757bffff3088c2e95cf (diff) | |
download | armnn-841aca155b35cc17ea9527599d2c364695e28166.tar.gz |
IVGCVSW-5200 Update pyarmnn
* Add HalfPixelCenters to Resize
* Update pyarmnn version to semantic versioning
* Add fill operator
* Add Bf16 optimization
* Add Gather operator
* Update TransposeConvolution2d descriptor
* Add Rank operator
* Add load dynamic tensor support of TfLiteParser
Signed-off-by: Jan Eilers <jan.eilers@arm.com>
Change-Id: I7e76ed286ab87bd97a65ff62868ba7db7967376f
Diffstat (limited to 'python/pyarmnn/test/test_tflite_parser.py')
-rw-r--r-- | python/pyarmnn/test/test_tflite_parser.py | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/python/pyarmnn/test/test_tflite_parser.py b/python/pyarmnn/test/test_tflite_parser.py index 344ec7ca13..8735eef3fb 100644 --- a/python/pyarmnn/test/test_tflite_parser.py +++ b/python/pyarmnn/test/test_tflite_parser.py @@ -7,6 +7,12 @@ import pyarmnn as ann import numpy as np +def test_TfLiteParserOptions_default_values(): + parserOptions = ann.TfLiteParserOptions() + assert parserOptions.m_InferAndValidate == False + assert parserOptions.m_StandInLayerForUnsupported == False + + @pytest.fixture() def parser(shared_data_folder): """ @@ -30,6 +36,42 @@ def test_check_tflite_parser_swig_ownership(parser): assert parser.thisown +def test_tflite_parser_with_optional_options(): + parserOptions = ann.TfLiteParserOptions() + parserOptions.m_InferAndValidate = True + parser = ann.ITfLiteParser(parserOptions) + assert parser.thisown + + +def create_with_opt() : + parserOptions = ann.TfLiteParserOptions() + parserOptions.m_InferAndValidate = True + return ann.ITfLiteParser(parserOptions) + +def test_tflite_parser_with_optional_options_out_of_scope(shared_data_folder): + parser = create_with_opt() + network = parser.CreateNetworkFromBinaryFile(os.path.join(shared_data_folder, "mock_model.tflite")) + + graphs_count = parser.GetSubgraphCount() + graph_id = graphs_count - 1 + + input_names = parser.GetSubgraphInputTensorNames(graph_id) + input_binding_info = parser.GetNetworkInputBindingInfo(graph_id, input_names[0]) + + output_names = parser.GetSubgraphOutputTensorNames(graph_id) + + preferred_backends = [ann.BackendId('CpuAcc'), ann.BackendId('CpuRef')] + + options = ann.CreationOptions() + runtime = ann.IRuntime(options) + + opt_network, messages = ann.Optimize(network, preferred_backends, runtime.GetDeviceSpec(), ann.OptimizerOptions()) + assert 0 == len(messages) + + net_id, messages = runtime.LoadNetwork(opt_network) + assert "" == messages + + def test_tflite_get_sub_graph_count(parser): graphs_count = parser.GetSubgraphCount() assert graphs_count == 1 |