Searched refs:splitTensorInfo2 (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/armnn/src/armnnTestUtils/ |
H A D | CreateWorkload.hpp | 1634 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()
|