diff options
author | Nikhil Raj <nikhil.raj@arm.com> | 2021-04-19 16:59:48 +0100 |
---|---|---|
committer | Nikhil Raj <nikhil.raj@arm.com> | 2021-04-27 17:37:11 +0100 |
commit | 5d955cf70ae0c5558d4f431f0fc6bd4552cd43a5 (patch) | |
tree | 4fb59200899808b8b008d6f48322d0d799b8b631 /src/armnnTfParser/test/Split.cpp | |
parent | 4a621c43174b6bdd9dc0bff839b245bc2139d6a6 (diff) | |
download | armnn-5d955cf70ae0c5558d4f431f0fc6bd4552cd43a5.tar.gz |
IVGCVSW-5721 Remove the Tensorflow Parser from ArmNN
Signed-off-by: Nikhil Raj <nikhil.raj@arm.com>
Change-Id: Ida37d3ee3a1af0c75aa905199bd861726c646846
Diffstat (limited to 'src/armnnTfParser/test/Split.cpp')
-rw-r--r-- | src/armnnTfParser/test/Split.cpp | 398 |
1 files changed, 0 insertions, 398 deletions
diff --git a/src/armnnTfParser/test/Split.cpp b/src/armnnTfParser/test/Split.cpp deleted file mode 100644 index eeef90a625..0000000000 --- a/src/armnnTfParser/test/Split.cpp +++ /dev/null @@ -1,398 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// - -#include "armnnTfParser/ITfParser.hpp" -#include "ParserPrototxtFixture.hpp" - -#include <armnn/utility/IgnoreUnused.hpp> - -#include <boost/test/unit_test.hpp> - -BOOST_AUTO_TEST_SUITE(TensorflowParser) - -struct SplitFixture : public armnnUtils::ParserPrototxtFixture<armnnTfParser::ITfParser> -{ - SplitFixture(bool withDimZero=false) { - m_Prototext = R"( - node { - name: "graphInput" - op: "Placeholder" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } - attr { - key: "shape" - value { - shape { - } - } - } - } - node { - name: "graphInput2" - op: "Placeholder" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } - attr { - key: "shape" - value { - shape { - } - } - } - } - node { - name: "multiplication" - op : "Mul" - input: "graphInput" - input: "graphInput2" - attr { - key: "T" - value { - type: DT_FLOAT - } - } - } - node { - name: "SplitInput" - op: "Const" - attr { - key: "dtype" - value { - type: DT_INT32 - } - } - attr { - key: "value" - value { - tensor { - dtype: DT_INT32 - tensor_shape { - } - int_val: )"; - - if(withDimZero) - { - m_Prototext += std::to_string(3); - } - else - { - m_Prototext += std::to_string(1); - } - - m_Prototext += R"( - } - } - } - } - node { - name: "Split" - op: "Split" )"; - if(withDimZero) - { - m_Prototext += "input: \"SplitInput\"\n"; - m_Prototext += "input: \"multiplication\"\n"; - } - else - { - m_Prototext += "input: \"graphInput\"\n"; - m_Prototext += "input: \"SplitInput\"\n"; - } - m_Prototext += R"( - attr { - key: "num_split" - value { - i: 2 - } - } - } - node { - name: "Relu_1" - op: "Relu" - input: "Split:0" - attr { - key: "T" - value { - type: DT_FLOAT - } - } - } - node { - name: "Relu_2" - op: "Relu" - input:"Split:1" - attr { - key: "T" - value { - type: DT_FLOAT - } - } - } )"; - - Setup( { { "graphInput", { 1, 2, 2 , 2} } , { "graphInput2", { 1, 2, 2 , 2} }}, - { "Relu_1", "Relu_2" }); - } -}; - -struct InputFirstSplitFixture : SplitFixture -{ - InputFirstSplitFixture() : SplitFixture(true) {} -}; - -BOOST_FIXTURE_TEST_CASE(ParseAxisOneSplitTwo, SplitFixture) -{ - BOOST_TEST( - (m_Parser->GetNetworkOutputBindingInfo("Relu_1").second.GetShape() == armnn::TensorShape({ 1, 1, 2, 2 }))); - - BOOST_TEST( - (m_Parser->GetNetworkOutputBindingInfo("Relu_2").second.GetShape() == armnn::TensorShape({ 1, 1, 2, 2 }))); - - RunTest<4>({ { "graphInput", { -1.0f, -0.5f, 1.25f, -3.0f, 0.0f, 0.5f, -0.75f, 1.75f } } }, - { { "Relu_1", { 0.0f, 0.0f, 1.25f, 0.0f } }, - { "Relu_2", { 0.0f, 0.5f, 0.0f, 1.75f } } }); -} - -BOOST_FIXTURE_TEST_CASE(ParseSplit, InputFirstSplitFixture) -{ - - BOOST_TEST( - (m_Parser->GetNetworkOutputBindingInfo("Relu_1").second.GetShape() == armnn::TensorShape({ 1, 2, 2, 1 }))); - - BOOST_TEST( - (m_Parser->GetNetworkOutputBindingInfo("Relu_2").second.GetShape() == armnn::TensorShape({ 1, 2, 2, 1 }))); - - RunTest<4>({ { "graphInput", { -1.0f, -0.5f, 1.25f, -3.0f, 0.0f, 0.5f, -0.75f , 1.75f } } , - { "graphInput2", { -1.0f, -0.5f, 1.25f, -3.0f, 0.0f, 0.5f, -0.75f , 1.75f } } }, - { { "Relu_1", { 1.0f, 1.5625f, 0, 0.5625f } }, - { "Relu_2", { 0.25, 9.0f, 0.25f, 3.0625f } } }); -} - -struct SplitLastDimFixture : public armnnUtils::ParserPrototxtFixture<armnnTfParser::ITfParser> -{ - SplitLastDimFixture(bool withDimZero=false) { - armnn::IgnoreUnused(withDimZero); - m_Prototext = R"( - node { - name: "Placeholder" - op: "Placeholder" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } - attr { - key: "shape" - value { - shape { - dim { - size: 1 - } - dim { - size: 2 - } - dim { - size: 2 - } - dim { - size: 3 - } - } - } - } - } - node { - name: "Const" - op: "Const" - attr { - key: "dtype" - value { - type: DT_INT32 - } - } - attr { - key: "value" - value { - tensor { - dtype: DT_INT32 - tensor_shape { - } - int_val: 3 - } - } - } - } - node { - name: "split/split_dim" - op: "Const" - attr { - key: "dtype" - value { - type: DT_INT32 - } - } - attr { - key: "value" - value { - tensor { - dtype: DT_INT32 - tensor_shape { - } - int_val: 3 - } - } - } - } - node { - name: "split" - op: "Split" - input: "split/split_dim" - input: "Placeholder" - attr { - key: "T" - value { - type: DT_FLOAT - } - } - attr { - key: "num_split" - value { - i: 3 - } - } - } - node { - name: "sub0/y" - op: "Const" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } - attr { - key: "value" - value { - tensor { - dtype: DT_FLOAT - tensor_shape { - } - float_val: 3.0 - } - } - } - } - node { - name: "sub0" - op: "Sub" - input: "split" - input: "sub0/y" - attr { - key: "T" - value { - type: DT_FLOAT - } - } - } - node { - name: "sub1/y" - op: "Const" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } - attr { - key: "value" - value { - tensor { - dtype: DT_FLOAT - tensor_shape { - } - float_val: 2.0 - } - } - } - } - node { - name: "sub1" - op: "Sub" - input: "split:1" - input: "sub1/y" - attr { - key: "T" - value { - type: DT_FLOAT - } - } - } - node { - name: "sub2/y" - op: "Const" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } - attr { - key: "value" - value { - tensor { - dtype: DT_FLOAT - tensor_shape { - } - float_val: 1.0 - } - } - } - } - node { - name: "sub2" - op: "Sub" - input: "split:2" - input: "sub2/y" - attr { - key: "T" - value { - type: DT_FLOAT - } - } - } - versions { - producer: 27 - } )"; - - Setup( { { "Placeholder", { 1, 2, 2 , 3} } }, - { "sub0", "sub1", "sub2" }); - } -}; - -BOOST_FIXTURE_TEST_CASE(SplitLastDimTest, SplitLastDimFixture) -{ - BOOST_TEST( - (m_Parser->GetNetworkOutputBindingInfo("sub0").second.GetShape() == armnn::TensorShape({ 1, 2, 2, 1 }))); - - BOOST_TEST( - (m_Parser->GetNetworkOutputBindingInfo("sub1").second.GetShape() == armnn::TensorShape({ 1, 2, 2, 1 }))); - - BOOST_TEST( - (m_Parser->GetNetworkOutputBindingInfo("sub2").second.GetShape() == armnn::TensorShape({ 1, 2, 2, 1 }))); - - RunTest<4>({ { "Placeholder", { 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f } } }, - { { "sub0", { -2.0f, 1.0f, 4.0f, 7.0f } }, - { "sub1", { 0.0f, 3.0f, 6.0f, 9.0f } }, - { "sub2", { 2.0f, 5.0f, 8.0f, 11.0f } } }); -} - -BOOST_AUTO_TEST_SUITE_END() |