aboutsummaryrefslogtreecommitdiff
path: root/src/armnnDeserializer/test
diff options
context:
space:
mode:
authorNattapat Chaimanowong <nattapat.chaimanowong@arm.com>2019-02-26 17:24:13 +0000
committernattapat.chaimanowong <nattapat.chaimanowong@arm.com>2019-02-27 16:39:47 +0000
commit6b4ed983aec66a979fbf206291d387bad0eeec9d (patch)
treecd44db489bbbdbf244f49f5ade3a346aad51a36f /src/armnnDeserializer/test
parentad402709dc137a001cab024cd3b4ad97132a2bf0 (diff)
downloadarmnn-6b4ed983aec66a979fbf206291d387bad0eeec9d.tar.gz
IVGCVSW-2682 Add Serializer and Deserializer for BatchToSpaceNd
Change-Id: I8b8ac99b4ecf282b796a8af4b30594ade4e061e6 Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Diffstat (limited to 'src/armnnDeserializer/test')
-rw-r--r--src/armnnDeserializer/test/DeserializeBatchToSpaceNd.cpp136
1 files changed, 136 insertions, 0 deletions
diff --git a/src/armnnDeserializer/test/DeserializeBatchToSpaceNd.cpp b/src/armnnDeserializer/test/DeserializeBatchToSpaceNd.cpp
new file mode 100644
index 0000000000..e85f4ada9d
--- /dev/null
+++ b/src/armnnDeserializer/test/DeserializeBatchToSpaceNd.cpp
@@ -0,0 +1,136 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#include <boost/test/unit_test.hpp>
+#include "ParserFlatbuffersSerializeFixture.hpp"
+#include "../Deserializer.hpp"
+
+#include <string>
+
+BOOST_AUTO_TEST_SUITE(Deserializer)
+
+struct BatchToSpaceNdFixture : public ParserFlatbuffersSerializeFixture
+{
+ explicit BatchToSpaceNdFixture(const std::string &inputShape,
+ const std::string &blockShape,
+ const std::string &crops,
+ const std::string &dataLayout,
+ const std::string &outputShape,
+ const std::string &dataType)
+ {
+ m_JsonString = R"(
+ {
+ inputIds: [0],
+ outputIds: [2],
+ layers: [
+ {
+ layer_type: "InputLayer",
+ layer: {
+ base: {
+ layerBindingId: 0,
+ base: {
+ index: 0,
+ layerName: "InputLayer",
+ layerType: "Input",
+ inputSlots: [{
+ index: 0,
+ connection: {sourceLayerIndex:0, outputSlotIndex:0 },
+ }],
+ outputSlots: [{
+ index: 0,
+ tensorInfo: {
+ dimensions: )" + inputShape + R"(,
+ dataType: )" + dataType + R"(
+ }
+ }]
+ }
+ }
+ }
+ },
+ {
+ layer_type: "BatchToSpaceNdLayer",
+ layer: {
+ base: {
+ index: 1,
+ layerName: "BatchToSpaceNdLayer",
+ layerType: "BatchToSpaceNd",
+ inputSlots: [{
+ index: 0,
+ connection: {sourceLayerIndex:0, outputSlotIndex:0 },
+ }],
+ outputSlots: [{
+ index: 0,
+ tensorInfo: {
+ dimensions: )" + outputShape + R"(,
+ dataType: )" + dataType + R"(
+ }
+ }]
+ },
+ descriptor: {
+ blockShape: )" + blockShape + R"(,
+ crops: )" + crops + R"(,
+ dataLayout: )" + dataLayout + R"(,
+ }
+ }
+ },
+ {
+ layer_type: "OutputLayer",
+ layer: {
+ base:{
+ layerBindingId: 2,
+ base: {
+ index: 2,
+ layerName: "OutputLayer",
+ layerType: "Output",
+ inputSlots: [{
+ index: 0,
+ connection: {sourceLayerIndex:1, outputSlotIndex:0 },
+ }],
+ outputSlots: [{
+ index: 0,
+ tensorInfo: {
+ dimensions: )" + outputShape + R"(,
+ dataType: )" + dataType + R"(
+ },
+ }],
+ }
+ }
+ },
+ }
+ ]
+ }
+ )";
+ SetupSingleInputSingleOutput("InputLayer", "OutputLayer");
+ }
+};
+
+struct SimpleBatchToSpaceNdFixture : BatchToSpaceNdFixture
+{
+ SimpleBatchToSpaceNdFixture() : BatchToSpaceNdFixture("[ 4, 2, 2, 1 ]",
+ "[ 2, 2 ]",
+ "[ 0, 0, 0, 0 ]",
+ "NHWC",
+ "[ 1, 4, 4, 1 ]",
+ "Float32") {}
+};
+
+BOOST_FIXTURE_TEST_CASE(SimpleBatchToSpaceNdFloat32, SimpleBatchToSpaceNdFixture)
+{
+ RunTest<4, armnn::DataType::Float32>(0,
+ {
+ 1.0f, 3.0f, 9.0f, 11.0f,
+ 2.0f, 4.0f, 10.0f, 12.0f,
+ 5.0f, 7.0f, 13.0f, 15.0f,
+ 6.0f, 8.0f, 14.0f, 16.0f
+ },
+ {
+ 1.0f, 2.0f, 3.0f, 4.0f,
+ 5.0f, 6.0f, 7.0f, 8.0f,
+ 9.0f, 10.0f, 11.0f, 12.0f,
+ 13.0f, 14.0f, 15.0f, 16.0f
+ });
+}
+
+BOOST_AUTO_TEST_SUITE_END()