From 5d955cf70ae0c5558d4f431f0fc6bd4552cd43a5 Mon Sep 17 00:00:00 2001 From: Nikhil Raj Date: Mon, 19 Apr 2021 16:59:48 +0100 Subject: IVGCVSW-5721 Remove the Tensorflow Parser from ArmNN Signed-off-by: Nikhil Raj Change-Id: Ida37d3ee3a1af0c75aa905199bd861726c646846 --- src/armnnTfParser/test/Split.cpp | 398 --------------------------------------- 1 file changed, 398 deletions(-) delete mode 100644 src/armnnTfParser/test/Split.cpp (limited to 'src/armnnTfParser/test/Split.cpp') 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 - -#include - -BOOST_AUTO_TEST_SUITE(TensorflowParser) - -struct SplitFixture : public armnnUtils::ParserPrototxtFixture -{ - 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 -{ - 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() -- cgit v1.2.1