ArmNN
 21.02
Unpack.cpp File Reference
#include <boost/test/unit_test.hpp>
#include "ParserFlatbuffersFixture.hpp"
#include "../TfLiteParser.hpp"
#include <string>
#include <iostream>

Go to the source code of this file.

Functions

 BOOST_FIXTURE_TEST_CASE (UnpackAxisZeroNumIsDefaultNotSpecified, DefaultUnpackAxisZeroFixture)
 
 BOOST_FIXTURE_TEST_CASE (UnpackAxisZeroNumIsDefaultNotSpecifiedUint8, DefaultUnpackAxisZeroUint8Fixture)
 
 BOOST_FIXTURE_TEST_CASE (UnpackLastAxisNumSix, DefaultUnpackLastAxisFixture)
 
 BOOST_FIXTURE_TEST_CASE (UnpackLastAxisNumSixUint8, DefaultUnpackLastAxisUint8Fixture)
 

Function Documentation

◆ BOOST_FIXTURE_TEST_CASE() [1/4]

BOOST_FIXTURE_TEST_CASE ( UnpackAxisZeroNumIsDefaultNotSpecified  ,
DefaultUnpackAxisZeroFixture   
)

Definition at line 113 of file Unpack.cpp.

114 {
115  RunTest<2, armnn::DataType::Float32>(
116  0,
117  { {"inputTensor", { 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f,
118  7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f,
119  13.0f, 14.0f, 15.0f, 16.0f, 17.0f, 18.0f,
120  19.0f, 20.0f, 21.0f, 22.0f, 23.0f, 24.0f } } },
121  { {"outputTensor1", { 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f }},
122  {"outputTensor2", { 7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f }},
123  {"outputTensor3", { 13.0f, 14.0f, 15.0f, 16.0f, 17.0f, 18.0f }},
124  {"outputTensor4", { 19.0f, 20.0f, 21.0f, 22.0f, 23.0f, 24.0f }} });
125 }

◆ BOOST_FIXTURE_TEST_CASE() [2/4]

BOOST_FIXTURE_TEST_CASE ( UnpackAxisZeroNumIsDefaultNotSpecifiedUint8  ,
DefaultUnpackAxisZeroUint8Fixture   
)

Definition at line 127 of file Unpack.cpp.

128 {
129  RunTest<2, armnn::DataType::QAsymmU8>(
130  0,
131  { {"inputTensor", { 1, 2, 3, 4, 5, 6,
132  7, 8, 9, 10, 11, 12,
133  13, 14, 15, 16, 17, 18,
134  19, 20, 21, 22, 23, 24 } } },
135  { {"outputTensor1", { 10, 20, 30, 40, 50, 60 }},
136  {"outputTensor2", { 70, 80, 90, 100, 110, 120 }},
137  {"outputTensor3", { 130, 140, 150, 160, 170, 180 }},
138  {"outputTensor4", { 190, 200, 210, 220, 230, 240 }} });
139 }

◆ BOOST_FIXTURE_TEST_CASE() [3/4]

BOOST_FIXTURE_TEST_CASE ( UnpackLastAxisNumSix  ,
DefaultUnpackLastAxisFixture   
)

Definition at line 151 of file Unpack.cpp.

152 {
153  RunTest<2, armnn::DataType::Float32>(
154  0,
155  { {"inputTensor", { 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f,
156  7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f,
157  13.0f, 14.0f, 15.0f, 16.0f, 17.0f, 18.0f,
158  19.0f, 20.0f, 21.0f, 22.0f, 23.0f, 24.0f } } },
159  { {"outputTensor1", { 1.0f, 7.0f, 13.0f, 19.0f }},
160  {"outputTensor2", { 2.0f, 8.0f, 14.0f, 20.0f }},
161  {"outputTensor3", { 3.0f, 9.0f, 15.0f, 21.0f }},
162  {"outputTensor4", { 4.0f, 10.0f, 16.0f, 22.0f }},
163  {"outputTensor5", { 5.0f, 11.0f, 17.0f, 23.0f }},
164  {"outputTensor6", { 6.0f, 12.0f, 18.0f, 24.0f }} });
165 }

◆ BOOST_FIXTURE_TEST_CASE() [4/4]

BOOST_FIXTURE_TEST_CASE ( UnpackLastAxisNumSixUint8  ,
DefaultUnpackLastAxisUint8Fixture   
)

Definition at line 167 of file Unpack.cpp.

References BOOST_AUTO_TEST_SUITE_END().

167  {
168  RunTest<2, armnn::DataType::QAsymmU8>(
169  0,
170  {{"inputTensor", { 1, 2, 3, 4, 5, 6,
171  7, 8, 9, 10, 11, 12,
172  13, 14, 15, 16, 17, 18,
173  19, 20, 21, 22, 23, 24 }}},
174  {{"outputTensor1", { 10, 70, 130, 190 }},
175  {"outputTensor2", { 20, 80, 140, 200 }},
176  {"outputTensor3", { 30, 90, 150, 210 }},
177  {"outputTensor4", { 40, 100, 160, 220 }},
178  {"outputTensor5", { 50, 110, 170, 230 }},
179  {"outputTensor6", { 60, 120, 180, 240 }}});
180 }