diff options
Diffstat (limited to 'src/armnnUtils')
-rw-r--r-- | src/armnnUtils/ParserPrototxtFixture.hpp | 20 | ||||
-rw-r--r-- | src/armnnUtils/PrototxtConversions.cpp | 32 | ||||
-rw-r--r-- | src/armnnUtils/PrototxtConversions.hpp | 16 | ||||
-rw-r--r-- | src/armnnUtils/test/PrototxtConversionsTest.cpp (renamed from src/armnnUtils/test/ParsePrototxtFixtureTest.cpp) | 11 |
4 files changed, 53 insertions, 26 deletions
diff --git a/src/armnnUtils/ParserPrototxtFixture.hpp b/src/armnnUtils/ParserPrototxtFixture.hpp index 154f6bec2a..be35e460cf 100644 --- a/src/armnnUtils/ParserPrototxtFixture.hpp +++ b/src/armnnUtils/ParserPrototxtFixture.hpp @@ -9,8 +9,6 @@ #include <test/TensorHelpers.hpp> -#include <armnnOnnxParser/IOnnxParser.hpp> - #include <Network.hpp> #include <VerificationHelpers.hpp> @@ -61,9 +59,6 @@ struct ParserPrototxtFixture void RunTest(const std::map<std::string, std::vector<float>>& inputData, const std::map<std::string, std::vector<float>>& expectedOutputData); - /// Converts an int value into the Protobuf octal representation - std::string ConvertInt32ToOctalString(int value); - std::string m_Prototext; std::unique_ptr<TParser, void(*)(TParser* parser)> m_Parser; armnn::IRuntimePtr m_Runtime; @@ -253,19 +248,4 @@ void ParserPrototxtFixture<TParser>::RunTest(const std::map<std::string, std::ve } } -template<typename TParser> -std::string ParserPrototxtFixture<TParser>::ConvertInt32ToOctalString(int value) -{ - std::stringstream ss; - std::string returnString; - for (int i = 0; i < 4; ++i) - { - ss << "\\"; - ss << std::setw(3) << std::setfill('0') << std::oct << ((value >> (i * 8)) & 0xFF); - } - - ss >> returnString; - return returnString; -} - } // namespace armnnUtils diff --git a/src/armnnUtils/PrototxtConversions.cpp b/src/armnnUtils/PrototxtConversions.cpp new file mode 100644 index 0000000000..1a6c053355 --- /dev/null +++ b/src/armnnUtils/PrototxtConversions.cpp @@ -0,0 +1,32 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#include "PrototxtConversions.hpp" + +#include <boost/format.hpp> + +#include <iomanip> +#include <sstream> +#include <string> + +namespace armnnUtils +{ + +/// Converts an int value into the Prototxt octal representation +std::string ConvertInt32ToOctalString(int value) +{ + std::stringstream ss; + std::string returnString; + for (int i = 0; i < 4; ++i) + { + ss << "\\"; + ss << std::setw(3) << std::setfill('0') << std::oct << ((value >> (i * 8)) & 0xFF); + } + + ss >> returnString; + return returnString; +} + +} // namespace armnnUtils diff --git a/src/armnnUtils/PrototxtConversions.hpp b/src/armnnUtils/PrototxtConversions.hpp new file mode 100644 index 0000000000..c90af9edff --- /dev/null +++ b/src/armnnUtils/PrototxtConversions.hpp @@ -0,0 +1,16 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#pragma once + +#include <string> + +namespace armnnUtils +{ + +/// Converts an int value into the Prototxt octal representation +std::string ConvertInt32ToOctalString(int value); + +} // namespace armnnUtils diff --git a/src/armnnUtils/test/ParsePrototxtFixtureTest.cpp b/src/armnnUtils/test/PrototxtConversionsTest.cpp index 926658ed0c..e06fbe0f2e 100644 --- a/src/armnnUtils/test/ParsePrototxtFixtureTest.cpp +++ b/src/armnnUtils/test/PrototxtConversionsTest.cpp @@ -3,17 +3,16 @@ // SPDX-License-Identifier: MIT // -#include <ParserPrototxtFixture.hpp> +#include <PrototxtConversions.hpp> #include <boost/test/unit_test.hpp> -#include "armnnTfParser/ITfParser.hpp" -BOOST_AUTO_TEST_SUITE(ParsePrototxtFixtureSuite) +BOOST_AUTO_TEST_SUITE(PrototxtConversions) -using Fixture = armnnUtils::ParserPrototxtFixture<armnnTfParser::ITfParser>; - -BOOST_FIXTURE_TEST_CASE(ConvertInt32ToOctalStringTest, Fixture) +BOOST_AUTO_TEST_CASE(ConvertInt32ToOctalStringTest) { + using armnnUtils::ConvertInt32ToOctalString; + std::string octalString = ConvertInt32ToOctalString(1); BOOST_ASSERT(octalString.compare("\\\\001\\\\000\\\\000\\\\000")); |