diff options
| author | Conor Kennedy <conor.kennedy@arm.com> | 2018-12-05 11:05:54 +0000 |
|---|---|---|
| committer | Aron Virginas-Tar <aron.virginas-tar@arm.com> | 2018-12-05 13:36:01 +0000 |
| commit | c2130a070e6a9196d193c93a02b5f118810dd59a (patch) | |
| tree | 45994da8a86619e4ed942a6619df284954329a1f /src/armnnTfParser/test/ExpandDims.cpp | |
| parent | f6ba747c0802d87ba30aecd598f0603f9bd18576 (diff) | |
| download | armnn-c2130a070e6a9196d193c93a02b5f118810dd59a.tar.gz | |
IVGCVSW-2193 ExpandDims operation implementation
* Add ExpandDims operation to TfParser.cpp
Change-Id: Ifa756ae0667c11e3b6daec8f6dd4e54cac88d16a
Diffstat (limited to 'src/armnnTfParser/test/ExpandDims.cpp')
| -rw-r--r-- | src/armnnTfParser/test/ExpandDims.cpp | 112 |
1 files changed, 112 insertions, 0 deletions
diff --git a/src/armnnTfParser/test/ExpandDims.cpp b/src/armnnTfParser/test/ExpandDims.cpp new file mode 100644 index 0000000000..57d472d41d --- /dev/null +++ b/src/armnnTfParser/test/ExpandDims.cpp @@ -0,0 +1,112 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#include <boost/test/unit_test.hpp> +#include "armnnTfParser/ITfParser.hpp" +#include "ParserPrototxtFixture.hpp" + +BOOST_AUTO_TEST_SUITE(TensorflowParser) + +struct ExpandDimsFixture : public armnnUtils::ParserPrototxtFixture<armnnTfParser::ITfParser> +{ + ExpandDimsFixture(const std::string& expandDim) + { + m_Prototext = + "node { \n" + " name: \"graphInput\" \n" + " op: \"Placeholder\" \n" + " attr { \n" + " key: \"dtype\" \n" + " value { \n" + " type: DT_FLOAT \n" + " } \n" + " } \n" + " attr { \n" + " key: \"shape\" \n" + " value { \n" + " shape { \n" + " } \n" + " } \n" + " } \n" + " } \n" + "node { \n" + " name: \"ExpandDims\" \n" + " op: \"ExpandDims\" \n" + " input: \"graphInput\" \n" + " attr { \n" + " key: \"T\" \n" + " value { \n" + " type: DT_FLOAT \n" + " } \n" + " } \n" + " attr { \n" + " key: \"Tdim\" \n" + " value { \n"; + m_Prototext += "i:" + expandDim; + m_Prototext += + " } \n" + " } \n" + "} \n"; + + SetupSingleInputSingleOutput({ 2, 3, 5 }, "graphInput", "ExpandDims"); + } +}; + +struct ExpandZeroDim : ExpandDimsFixture +{ + ExpandZeroDim() : ExpandDimsFixture("0") {} +}; + +struct ExpandTwoDim : ExpandDimsFixture +{ + ExpandTwoDim() : ExpandDimsFixture("2") {} +}; + +struct ExpandThreeDim : ExpandDimsFixture +{ + ExpandThreeDim() : ExpandDimsFixture("3") {} +}; + +struct ExpandMinusOneDim : ExpandDimsFixture +{ + ExpandMinusOneDim() : ExpandDimsFixture("-1") {} +}; + +struct ExpandMinusThreeDim : ExpandDimsFixture +{ + ExpandMinusThreeDim() : ExpandDimsFixture("-3") {} +}; + +BOOST_FIXTURE_TEST_CASE(ParseExpandZeroDim, ExpandZeroDim) +{ + BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo("ExpandDims").second.GetShape() == + armnn::TensorShape({1, 2, 3, 5}))); +} + +BOOST_FIXTURE_TEST_CASE(ParseExpandTwoDim, ExpandTwoDim) +{ + BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo("ExpandDims").second.GetShape() == + armnn::TensorShape({2, 3, 1, 5}))); +} + +BOOST_FIXTURE_TEST_CASE(ParseExpandThreeDim, ExpandThreeDim) +{ + BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo("ExpandDims").second.GetShape() == + armnn::TensorShape({2, 3, 5, 1}))); +} + +BOOST_FIXTURE_TEST_CASE(ParseExpandMinusOneDim, ExpandMinusOneDim) +{ + BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo("ExpandDims").second.GetShape() == + armnn::TensorShape({2, 3, 5, 1}))); +} + +BOOST_FIXTURE_TEST_CASE(ParseExpandMinusThreeDim, ExpandMinusThreeDim) +{ + BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo("ExpandDims").second.GetShape() == + armnn::TensorShape({2, 1, 3, 5}))); +} + +BOOST_AUTO_TEST_SUITE_END() |