diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2023-05-09 13:02:04 +0100 |
---|---|---|
committer | Jeremy Johnson <jeremy.johnson@arm.com> | 2023-05-09 13:02:04 +0100 |
commit | 7037332896b3151963a3e83f65acb40590c74199 (patch) | |
tree | 37cc187bbad1ed0c52efa6f3a47e72cb24b3e89a /operators/tensor/argmax/argmax_5x9x7x7x7x8_i8_axis5/test.json | |
parent | fe6e6eb29a416ad7618567d4365f8b65165694c1 (diff) | |
download | conformance_tests-7037332896b3151963a3e83f65acb40590c74199.tar.gz |
8K level: rank 6 ARGMAX tests
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: Ia7c3d89b176a2ac13fd05c734a3d8b7753fa8409
Diffstat (limited to 'operators/tensor/argmax/argmax_5x9x7x7x7x8_i8_axis5/test.json')
-rw-r--r-- | operators/tensor/argmax/argmax_5x9x7x7x7x8_i8_axis5/test.json | 64 |
1 files changed, 64 insertions, 0 deletions
diff --git a/operators/tensor/argmax/argmax_5x9x7x7x7x8_i8_axis5/test.json b/operators/tensor/argmax/argmax_5x9x7x7x7x8_i8_axis5/test.json new file mode 100644 index 000000000..5689f9f9b --- /dev/null +++ b/operators/tensor/argmax/argmax_5x9x7x7x7x8_i8_axis5/test.json @@ -0,0 +1,64 @@ +{ + version: { + _major: 0, + _minor: 70, + _patch: 0, + _draft: true + }, + regions: [ + { + name: "main", + blocks: [ + { + name: "main", + operators: [ + { + op: "ARGMAX", + attribute_type: "AxisAttribute", + attribute: { + axis: 5 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 5, + 9, + 7, + 7, + 7, + 8 + ], + type: "INT8" + }, + { + name: "result-0", + shape: [ + 5, + 9, + 7, + 7, + 7 + ], + type: "INT32" + } + ], + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ] + } + ] +} |