MCPcopy Create free account
hub / github.com/ARM-software/armnn / ReshapeUint8Test

Function ReshapeUint8Test

delegate/test/ReshapeTest.cpp:177–203  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

175}
176
177void ReshapeUint8Test(bool useOption = true, const std::vector<armnn::BackendId>& backends = {})
178{
179 // Set input data
180 std::vector<int32_t> inputShape { 1, 3, 4, 1 };
181 std::vector<int32_t> outputShape { 6, 2 };
182 std::vector<int32_t> targetShape { -1, 2 };
183
184 std::vector<uint8_t> inputValues = { 5, 8, 10, 7,
185 8, 12, 15, 2,
186 3, 4, 1, 11 };
187
188 std::vector<uint8_t> expectedOutputValues = { 5, 8, 10, 7,
189 8, 12, 15, 2,
190 3, 4, 1, 11 };
191
192 RedefineTest<uint8_t>(tflite::BuiltinOperator_RESHAPE,
193 ::tflite::TensorType_UINT8,
194 inputShape,
195 outputShape,
196 inputValues,
197 expectedOutputValues,
198 targetShape,
199 useOption,
200 backends,
201 2.5f,
202 1);
203}
204
205void ReshapeInt16Test(bool useOption = true, const std::vector<armnn::BackendId>& backends = {})
206{

Callers 1

ReshapeTest.cppFile · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected