diff options
author | Matthew Sloyan <matthew.sloyan@arm.com> | 2021-04-15 20:46:24 +0100 |
---|---|---|
committer | Matthew Sloyan <matthew.sloyan@arm.com> | 2021-04-15 20:51:23 +0100 |
commit | ed7fce413410d15c501ea52f9e6bfbbf71b3daf1 (patch) | |
tree | f72645fdb44fec0fee95e00bd6c6f51b404ed5c7 /src/armnnTfLiteParser/test/Neg.cpp | |
parent | 86723e67a940ed9f288c3cb59c120b9a10a1d590 (diff) | |
download | armnn-ed7fce413410d15c501ea52f9e6bfbbf71b3daf1.tar.gz |
IVGCVSW-5829 Segfault in tflite-parser, int8 models
* Updated ParseSplit TfLiteParser function to read correct axis data.
* Improved validation in ParseSplit and ParseSplitV function.
* Added TensorFlow BOOL support to TfLiteParser.
* Added supported ElementWiseUnary operators to TfLiteParser
E.g. ABS, LOGICAL_NOT and RSQRT.
* Removed ParseExp and ParseNeg function implementation in favour
of reusable ParseElementWiseUnary function.
* Removed Exp.cpp and Neg.cpp files and moved tests to ElementWiseUnary.cpp.
Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Change-Id: Ibce36e3ce4d95755dda88abc2ddde1e07e62c5e2
Diffstat (limited to 'src/armnnTfLiteParser/test/Neg.cpp')
-rw-r--r-- | src/armnnTfLiteParser/test/Neg.cpp | 85 |
1 files changed, 0 insertions, 85 deletions
diff --git a/src/armnnTfLiteParser/test/Neg.cpp b/src/armnnTfLiteParser/test/Neg.cpp deleted file mode 100644 index 39e1f9e82c..0000000000 --- a/src/armnnTfLiteParser/test/Neg.cpp +++ /dev/null @@ -1,85 +0,0 @@ -// -// Copyright © 2020 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// - -#include <boost/test/unit_test.hpp> -#include "ParserFlatbuffersFixture.hpp" -#include "../TfLiteParser.hpp" - -#include <string> - -BOOST_AUTO_TEST_SUITE(TensorflowLiteParser) - -struct NegFixture : public ParserFlatbuffersFixture -{ - explicit NegFixture(const std::string & inputShape, - const std::string & outputShape) - { - m_JsonString = R"( - { - "version": 3, - "operator_codes": [ { "builtin_code": "NEG" } ], - "subgraphs": [ { - "tensors": [ - { - "shape": )" + inputShape + R"(, - "type": "FLOAT32", - "buffer": 0, - "name": "inputTensor", - "quantization": { - "min": [ 0.0 ], - "max": [ 255.0 ], - "scale": [ 1.0 ], - "zero_point": [ 0 ], - } - }, - { - "shape": )" + outputShape + R"( , - "type": "FLOAT32", - "buffer": 1, - "name": "outputTensor", - "quantization": { - "min": [ 0.0 ], - "max": [ 255.0 ], - "scale": [ 1.0 ], - "zero_point": [ 0 ], - } - } - ], - "inputs": [ 0 ], - "outputs": [ 1 ], - "operators": [ - { - "opcode_index": 0, - "inputs": [ 0 ], - "outputs": [ 1 ], - "custom_options_format": "FLEXBUFFERS" - } - ], - } ], - "buffers" : [ - { }, - { } - ] - } - )"; - Setup(); - } -}; - -struct SimpleNegFixture : public NegFixture -{ - SimpleNegFixture() : NegFixture("[ 1, 2, 3, 1 ]", "[ 1, 2, 3, 1 ]") {} -}; - -BOOST_FIXTURE_TEST_CASE(ParseNeg, SimpleNegFixture) -{ - using armnn::DataType; - RunTest<4, DataType::Float32>(0, {{ "inputTensor", { 0.0f, 1.0f, -2.0f, - 20.0855185f, -54.5980834f, 5.0f} }}, - {{ "outputTensor",{ 0.0f, -1.0f, 2.0f, - -20.0855185f, 54.5980834f, -5.0f} }}); -} - -BOOST_AUTO_TEST_SUITE_END() |