diff options
| author | Richard Burton <richard.burton@arm.com> | 2020-04-08 16:39:05 +0100 |
|---|---|---|
| committer | Jim Flynn <jim.flynn@arm.com> | 2020-04-10 16:11:09 +0000 |
| commit | dc0c6ed9f8b993e63f492f203d7d7080ab4c835c (patch) | |
| tree | ea8541990b13ebf1a038009aa6b8b4b1ea8c3f55 /python/pyarmnn/test/testdata/shared | |
| parent | fe5a24beeef6e9a41366e694f41093565e748048 (diff) | |
| download | armnn-dc0c6ed9f8b993e63f492f203d7d7080ab4c835c.tar.gz | |
Add PyArmNN to work with ArmNN API of 20.02
* Add Swig rules for generating python wrapper
* Add documentation
* Add tests and testing data
Change-Id: If48eda08931514fa21e72214dfead2835f07237c
Signed-off-by: Richard Burton <richard.burton@arm.com>
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
Diffstat (limited to 'python/pyarmnn/test/testdata/shared')
14 files changed, 226 insertions, 0 deletions
diff --git a/python/pyarmnn/test/testdata/shared/caffe_parser/golden_output_caffe.npy b/python/pyarmnn/test/testdata/shared/caffe_parser/golden_output_caffe.npy Binary files differnew file mode 100644 index 0000000000..007141cb9f --- /dev/null +++ b/python/pyarmnn/test/testdata/shared/caffe_parser/golden_output_caffe.npy diff --git a/python/pyarmnn/test/testdata/shared/caffe_parser/input_caffe.npy b/python/pyarmnn/test/testdata/shared/caffe_parser/input_caffe.npy Binary files differnew file mode 100644 index 0000000000..15df758b58 --- /dev/null +++ b/python/pyarmnn/test/testdata/shared/caffe_parser/input_caffe.npy diff --git a/python/pyarmnn/test/testdata/shared/license.txt b/python/pyarmnn/test/testdata/shared/license.txt new file mode 100644 index 0000000000..1e95a68e0d --- /dev/null +++ b/python/pyarmnn/test/testdata/shared/license.txt @@ -0,0 +1,10 @@ +This folder contains models and data needed for the testing of PyArmNN. + +All models and files found in this folder were created by ARM for the purpose +of testing PyArmNN. + +All the contents of this folder are distributed with the following license. + +Copyright © 2020 Arm Ltd. All rights reserved. +SPDX-License-Identifier: MIT + diff --git a/python/pyarmnn/test/testdata/shared/mock_model.caffemodel b/python/pyarmnn/test/testdata/shared/mock_model.caffemodel Binary files differnew file mode 100644 index 0000000000..df4079b729 --- /dev/null +++ b/python/pyarmnn/test/testdata/shared/mock_model.caffemodel diff --git a/python/pyarmnn/test/testdata/shared/mock_model.onnx b/python/pyarmnn/test/testdata/shared/mock_model.onnx Binary files differnew file mode 100644 index 0000000000..c1b506cc16 --- /dev/null +++ b/python/pyarmnn/test/testdata/shared/mock_model.onnx diff --git a/python/pyarmnn/test/testdata/shared/mock_model.pb b/python/pyarmnn/test/testdata/shared/mock_model.pb Binary files differnew file mode 100644 index 0000000000..cff9dc7add --- /dev/null +++ b/python/pyarmnn/test/testdata/shared/mock_model.pb diff --git a/python/pyarmnn/test/testdata/shared/mock_model.tflite b/python/pyarmnn/test/testdata/shared/mock_model.tflite Binary files differnew file mode 100644 index 0000000000..0b8944d3ed --- /dev/null +++ b/python/pyarmnn/test/testdata/shared/mock_model.tflite diff --git a/python/pyarmnn/test/testdata/shared/mock_profile_out.json b/python/pyarmnn/test/testdata/shared/mock_profile_out.json new file mode 100644 index 0000000000..8e1056160b --- /dev/null +++ b/python/pyarmnn/test/testdata/shared/mock_profile_out.json @@ -0,0 +1,216 @@ +{ + "ArmNN": { + "inference_measurements_#1": { + "type": "Event", + "Wall clock time_#1": { + "type": "Measurement", + "raw": [ + 1.1, + 2.2, + 3.3, + 4.4, + 5.5, + 6.6 + ], + "unit": "us" + }, + + "Execute_#2": { + "type": "Event", + "Wall clock time_#2": { + "type": "Measurement", + "raw": [ + 1.1, + 2.2, + 3.3, + 4.4, + 5.5, + 6.6 + ], + "unit": "us" + }, + "Wall clock time (Start)_#2": { + "type": "Measurement", + "raw": [ + 1, + 1, + 1, + 1, + 1, + 1 + ], + "unit": "us" + }, + "Wall clock time (Stop)_#2": { + "type": "Measurement", + "raw": [ + 2, + 2, + 2, + 2, + 2, + 2 + ], + "unit": "us" + }, + + "RefSomeMock1dWorkload_Execute_#5": { + "type": "Event", + "Wall clock time_#5": { + "type": "Measurement", + "raw": [ + 2, + 2, + 2, + 2, + 2, + 2 + ], + "unit": "us" + }, + "Wall clock time (Start)_#5": { + "type": "Measurement", + "raw": [ + 2, + 2, + 2, + 2, + 2, + 2 + ], + "unit": "us" + }, + "Wall clock time (Stop)_#5": { + "type": "Measurement", + "raw": [ + 4, + 4, + 4, + 4, + 4, + 4 + ], + "unit": "us" + } + }, + "NeonSomeMock2Workload_Execute_#6": { + "type": "Event", + "Wall clock time_#6": { + "type": "Measurement", + "raw": [ + 2, + 2, + 2, + 2, + 2, + 2 + ], + "unit": "us" + }, + "Wall clock time (Start)_#6": { + "type": "Measurement", + "raw": [ + 2, + 2, + 2, + 2, + 2, + 2 + ], + "unit": "us" + }, + "Wall clock time (Stop)_#6": { + "type": "Measurement", + "raw": [ + 4, + 4, + 4, + 4, + 4, + 4 + ], + "unit": "us" + } + }, + "ClSomeMock3dWorkload_Execute_#7": { + "type": "Event", + "Wall clock time_#7": { + "type": "Measurement", + "raw": [ + 2, + 2, + 2, + 2, + 2, + 2 + ], + "unit": "us" + }, + "Wall clock time (Start)_#7": { + "type": "Measurement", + "raw": [ + 2, + 2, + 2, + 2, + 2, + 2 + ], + "unit": "us" + }, + "Wall clock time (Stop)_#7": { + "type": "Measurement", + "raw": [ + 4, + 4, + 4, + 4, + 4, + 4 + ], + "unit": "us" + } + }, + "EthosNSomeMock4dWorkload_Execute_#8": { + "type": "Event", + "Wall clock time_#8": { + "type": "Measurement", + "raw": [ + 2, + 2, + 2, + 2, + 2, + 2 + ], + "unit": "us" + }, + "Wall clock time (Start)_#8": { + "type": "Measurement", + "raw": [ + 2, + 2, + 2, + 2, + 2, + 2 + ], + "unit": "us" + }, + "Wall clock time (Stop)_#8": { + "type": "Measurement", + "raw": [ + 4, + 4, + 4, + 4, + 4, + 4 + ], + "unit": "us" + } + } + } + } + } +} diff --git a/python/pyarmnn/test/testdata/shared/onnx_parser/golden_output_onnx.npy b/python/pyarmnn/test/testdata/shared/onnx_parser/golden_output_onnx.npy Binary files differnew file mode 100644 index 0000000000..f83d6ea7cb --- /dev/null +++ b/python/pyarmnn/test/testdata/shared/onnx_parser/golden_output_onnx.npy diff --git a/python/pyarmnn/test/testdata/shared/onnx_parser/input_onnx.npy b/python/pyarmnn/test/testdata/shared/onnx_parser/input_onnx.npy Binary files differnew file mode 100644 index 0000000000..15df758b58 --- /dev/null +++ b/python/pyarmnn/test/testdata/shared/onnx_parser/input_onnx.npy diff --git a/python/pyarmnn/test/testdata/shared/tf_parser/golden_output_tf.npy b/python/pyarmnn/test/testdata/shared/tf_parser/golden_output_tf.npy Binary files differnew file mode 100644 index 0000000000..007141cb9f --- /dev/null +++ b/python/pyarmnn/test/testdata/shared/tf_parser/golden_output_tf.npy diff --git a/python/pyarmnn/test/testdata/shared/tf_parser/input_tf.npy b/python/pyarmnn/test/testdata/shared/tf_parser/input_tf.npy Binary files differnew file mode 100644 index 0000000000..a21802e4b8 --- /dev/null +++ b/python/pyarmnn/test/testdata/shared/tf_parser/input_tf.npy diff --git a/python/pyarmnn/test/testdata/shared/tflite_parser/golden_output_lite.npy b/python/pyarmnn/test/testdata/shared/tflite_parser/golden_output_lite.npy Binary files differnew file mode 100644 index 0000000000..099f7fed22 --- /dev/null +++ b/python/pyarmnn/test/testdata/shared/tflite_parser/golden_output_lite.npy diff --git a/python/pyarmnn/test/testdata/shared/tflite_parser/input_lite.npy b/python/pyarmnn/test/testdata/shared/tflite_parser/input_lite.npy Binary files differnew file mode 100644 index 0000000000..53174683ff --- /dev/null +++ b/python/pyarmnn/test/testdata/shared/tflite_parser/input_lite.npy |