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

Function ConcatUint8TwoInputsTest

delegate/test/ControlTest.cpp:18–38  ·  view source on GitHub ↗

CONCATENATION Operator

Source from the content-addressed store, hash-verified

16
17// CONCATENATION Operator
18void ConcatUint8TwoInputsTest()
19{
20 std::vector<int32_t> inputShape { 2, 2 };
21 std::vector<int32_t> expectedOutputShape { 4, 2 };
22
23 // Set input and output data
24 std::vector<std::vector<uint8_t>> inputValues;
25 std::vector<uint8_t> inputValue1 { 0, 1, 2, 3 }; // Lower bounds
26 std::vector<uint8_t> inputValue2 { 252, 253, 254, 255 }; // Upper bounds
27 inputValues.push_back(inputValue1);
28 inputValues.push_back(inputValue2);
29
30 std::vector<uint8_t> expectedOutputValues { 0, 1, 2, 3, 252, 253, 254, 255 };
31
32 ConcatenationTest<uint8_t>(tflite::BuiltinOperator_CONCATENATION,
33 ::tflite::TensorType_UINT8,
34 inputShape,
35 expectedOutputShape,
36 inputValues,
37 expectedOutputValues);
38}
39
40void ConcatInt16TwoInputsTest()
41{

Callers 1

ControlTest.cppFile · 0.85

Calls 1

push_backMethod · 0.80

Tested by

no test coverage detected