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

Function SplitVFp32ZeroDimTest

delegate/test/SplitTest.cpp:164–189  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

162}
163
164void SplitVFp32ZeroDimTest()
165{
166 uint32_t inputShape[] = { 1 };
167
168 uint32_t outputShape0[] = { 1 };
169 uint32_t outputShape1[] = { 1 };
170 uint32_t outputShape2[] = { 1 };
171 uint32_t outputShape3[] = { 1 };
172 uint32_t outputShape4[] = { 1 };
173 uint32_t outputShape5[] = { 1 };
174 uint32_t outputShape6[] = { 2 };
175 uint32_t outputShape7[] = { 0 };
176
177 armnn::TensorInfo inputTensor = armnn::TensorInfo(1, inputShape, armnn::DataType::Float32);
178 armnn::TensorInfo outputTensor0 = armnn::TensorInfo(1, outputShape0, armnn::DataType::Float32);
179 armnn::TensorInfo outputTensor1 = armnn::TensorInfo(1, outputShape1, armnn::DataType::Float32);
180 armnn::TensorInfo outputTensor2 = armnn::TensorInfo(1, outputShape2, armnn::DataType::Float32);
181 armnn::TensorInfo outputTensor3 = armnn::TensorInfo(1, outputShape3, armnn::DataType::Float32);
182 armnn::TensorInfo outputTensor4 = armnn::TensorInfo(1, outputShape4, armnn::DataType::Float32);
183 armnn::TensorInfo outputTensor5 = armnn::TensorInfo(1, outputShape5, armnn::DataType::Float32);
184 armnn::TensorInfo outputTensor6 = armnn::TensorInfo(1, outputShape6, armnn::DataType::Float32);
185 armnn::TensorInfo outputTensor7 = armnn::TensorInfo(1, outputShape7, armnn::DataType::Float32);
186
187 CHECK(ZeroDimPresent({inputTensor, outputTensor0, outputTensor1, outputTensor2, outputTensor3,
188 outputTensor4, outputTensor5, outputTensor6, outputTensor7}) == true);
189}
190
191// SPLIT_V Test Suite
192TEST_SUITE("SPLIT_VTests")

Callers 1

SplitTest.cppFile · 0.85

Calls 2

ZeroDimPresentFunction · 0.85
TensorInfoClass · 0.50

Tested by

no test coverage detected