diff options
| author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-04-17 11:22:38 +0100 |
|---|---|---|
| committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-04-17 11:55:30 +0100 |
| commit | 4818d465973d72979a7f6e783e1b55b320781710 (patch) | |
| tree | 7c6cce52a728b6fe3ea2733b808555ff27edbab2 /src/armnnTfLiteParser/test/Minimum.cpp | |
| parent | 0790dcea1056298d63f97dec904c8ade5d21f439 (diff) | |
| download | armnn-4818d465973d72979a7f6e783e1b55b320781710.tar.gz | |
IVGCVSW-2849 Add TfLite Parser support for Rank-0 operands and unit tests
Change-Id: I6dab12aed395a30466d66421c6e5a12659fedac8
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Diffstat (limited to 'src/armnnTfLiteParser/test/Minimum.cpp')
| -rw-r--r-- | src/armnnTfLiteParser/test/Minimum.cpp | 77 |
1 files changed, 77 insertions, 0 deletions
diff --git a/src/armnnTfLiteParser/test/Minimum.cpp b/src/armnnTfLiteParser/test/Minimum.cpp index 70b249ff5f..8c6db680e7 100644 --- a/src/armnnTfLiteParser/test/Minimum.cpp +++ b/src/armnnTfLiteParser/test/Minimum.cpp @@ -173,4 +173,81 @@ BOOST_FIXTURE_TEST_CASE(ParseMinimumBroadcast1D4D, MinimumBroadcastFixture1D4D) 5.0f, 6.0f, 7.0f }}}); } +struct MinimumBroadcastFixture2D0D : public ParserFlatbuffersFixture +{ + explicit MinimumBroadcastFixture2D0D() + { + m_JsonString = R"( + { + "version": 3, + "operator_codes": [ { "builtin_code": "MINIMUM" } ], + "subgraphs": [ { + "tensors": [ + { + "shape": [ 1, 2 ], + "type": "FLOAT32", + "buffer": 0, + "name": "input0", + "quantization": { + "min": [ 0.0 ], + "max": [ 255.0 ], + "scale": [ 1.0 ], + "zero_point": [ 0 ], + } + }, + { + "shape": [ ], + "type": "FLOAT32", + "buffer": 2, + "name": "input1", + "quantization": { + "min": [ 0.0 ], + "max": [ 255.0 ], + "scale": [ 1.0 ], + "zero_point": [ 0 ], + } + }, + { + "shape": [ 1, 2 ] , + "type": "FLOAT32", + "buffer": 1, + "name": "output", + "quantization": { + "min": [ 0.0 ], + "max": [ 255.0 ], + "scale": [ 1.0 ], + "zero_point": [ 0 ], + } + } + ], + "inputs": [ 0 ], + "outputs": [ 2 ], + "operators": [ + { + "opcode_index": 0, + "inputs": [ 0, 1 ], + "outputs": [ 2 ], + "custom_options_format": "FLEXBUFFERS" + } + ], + } ], + "buffers" : [ + { }, + { }, + { "data": [ 0, 0, 0, 64 ] } + ] + } + )"; + Setup(); + } +}; + +BOOST_FIXTURE_TEST_CASE(ParseMinimumBroadcast2D0D, MinimumBroadcastFixture2D0D) +{ + RunTest<2, armnn::DataType::Float32>( + 0, + {{"input0", { 1.0f, 5.0f }}}, + {{"output", { 1.0f, 2.0f }}}); +} + BOOST_AUTO_TEST_SUITE_END() |