From bceff2fb3fc68bb0aa88b886900c34b77340c826 Mon Sep 17 00:00:00 2001 From: surmeh01 Date: Thu, 29 Mar 2018 16:29:27 +0100 Subject: Release 18.03 --- src/armnnTfParser/test/Softmax.cpp | 55 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 55 insertions(+) create mode 100644 src/armnnTfParser/test/Softmax.cpp (limited to 'src/armnnTfParser/test/Softmax.cpp') diff --git a/src/armnnTfParser/test/Softmax.cpp b/src/armnnTfParser/test/Softmax.cpp new file mode 100644 index 0000000000..1ab28ea3aa --- /dev/null +++ b/src/armnnTfParser/test/Softmax.cpp @@ -0,0 +1,55 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// See LICENSE file in the project root for full license information. +// + +#include +#include "armnnTfParser/ITfParser.hpp" +#include "ParserPrototxtFixture.hpp" + +BOOST_AUTO_TEST_SUITE(TensorflowParser) + +struct SoftmaxFixture : public ParserPrototxtFixture +{ + SoftmaxFixture() + { + m_Prototext = "node {\n" + " name: \"blah\"\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: \"blah2\"\n" + " op: \"Softmax\"\n" + " input: \"blah\"\n" + " attr {\n" + " key: \"T\"\n" + " value {\n" + " type: DT_FLOAT\n" + " }\n" + " }\n" + "}\n"; + + SetupSingleInputSingleOutput({ 1, 7 }, "blah", "blah2"); + } +}; + +BOOST_FIXTURE_TEST_CASE(ParseSoftmax, SoftmaxFixture) +{ + RunTest<2>({ 0, 0, 10000, 0, 0, 0, 0 }, { 0, 0, 1, 0, 0, 0, 0 }); +} + + +BOOST_AUTO_TEST_SUITE_END() -- cgit v1.2.1