Home
last modified time | relevance | path

Searched refs:splitTensorInfo2 (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/armnn/src/armnnTestUtils/
H A DCreateWorkload.hpp1634 armnn::TensorInfo splitTensorInfo2({ 1, 1, 100, 10 }, DataType); in CreateSplitterConcatWorkloadTest() local
1677 Connect(splitter, concat, splitTensorInfo2, 1, 0); // So that the outputs are flipped round. in CreateSplitterConcatWorkloadTest()
1707 armnn::TensorInfo splitTensorInfo2({ 1, 2, 100, 50 }, DataType); in CreateSplitterMultipleInputsOneOutputWorkloadTest() local
1743 Connect(splitter, activ1_0, splitTensorInfo2, 1, 0); in CreateSplitterMultipleInputsOneOutputWorkloadTest()
1744 Connect(splitter, activ1_1, splitTensorInfo2, 1, 0); in CreateSplitterMultipleInputsOneOutputWorkloadTest()
1748 Connect(activ1_0, output3, splitTensorInfo2, 0, 0); in CreateSplitterMultipleInputsOneOutputWorkloadTest()
1749 Connect(activ1_1, output4, splitTensorInfo2, 0, 0); in CreateSplitterMultipleInputsOneOutputWorkloadTest()