Searched defs:RHSInputShape (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/armnn/delegate/test/ |
| H A D | BatchMatMulTest.cpp | 22 std::vector<int32_t> RHSInputShape { 2, 2 }; in BatchMatMul2DFp32SimpleTest() local
50 std::vector<int32_t> RHSInputShape { 2, 2 }; in BatchMatMul2DInt8SimpleTest() local
79 std::vector<int32_t> RHSInputShape { 1,2,2 }; in BatchMatMul3DFp32SimpleTest() local
108 std::vector<int32_t> RHSInputShape { 1,2,2 }; in BatchMatMul3DInt8SimpleTest() local
137 std::vector<int32_t> RHSInputShape { 1,1,2,2 }; in BatchMatMul4DFp32SimpleTest() local
166 std::vector<int32_t> RHSInputShape { 1,1,2,2 }; in BatchMatMul4DInt8SimpleTest() local
195 std::vector<int32_t> RHSInputShape { 2,2,2 }; in BatchMatMul3DFp32BatchTest() local
233 std::vector<int32_t> RHSInputShape { 2,2,2 }; in BatchMatMul3DInt8BatchTest() local
271 std::vector<int32_t> RHSInputShape { 2,2 }; in BatchMatMul3DFp32BroadcastTest() local
306 std::vector<int32_t> RHSInputShape { 1,2,2 }; in BatchMatMul3DInt8BroadcastTest() local
[all …]
|
| H A D | BatchMatMulTestHelper.hpp | 125 std::vector<int32_t>& RHSInputShape, in BatchMatMulTest()
|