Searched defs:inputXInfo (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
| H A D | BatchMatMulTestImpl.cpp | 28 const armnn::TensorInfo& inputXInfo, in BatchMatMulTestImpl()
77 armnn::TensorInfo inputXInfo({2,2}, ArmnnType, qScale, qOffset); in BatchMatMul2DSimpleTest() local
155 armnn::TensorInfo inputXInfo({1,2,2}, ArmnnType, qScale, qOffset); in BatchMatMul3DSimpleTest() local
233 armnn::TensorInfo inputXInfo({1,1,2,2}, ArmnnType, qScale, qOffset); in BatchMatMulNCHWSimpleTest() local
316 armnn::TensorInfo inputXInfo({1,2,2,1}, ArmnnType, qScale, qOffset); in BatchMatMulNHWCSimpleTest() local
394 armnn::TensorInfo inputXInfo({2,2,2}, ArmnnType, qScale, qOffset); in BatchMatMul3DBatchTest() local
481 armnn::TensorInfo inputXInfo({2,2,2}, ArmnnType, qScale, qOffset); in BatchMatMul3DBroadcastTest() local
565 armnn::TensorInfo inputXInfo({2,2,2}, ArmnnType, qScale, qOffset); in BatchMatMul3D2DBroadcastTest() local
654 armnn::TensorInfo inputXInfo({1,1,2,2,2}, ArmnnType, qScale, qOffset); in BatchMatMulNDHWCNHWCTest() local
741 armnn::TensorInfo inputXInfo({1,1}, ArmnnType, qScale, qOffset); in BatchMatMul2DTinyTest() local
[all …]
|
| /aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
| H A D | BatchMatMulImpl.hpp | 35 TensorInfo inputXInfo; member in armnn::BatchMatMul
|
| H A D | RefBatchMatMulWorkload.cpp | 34 const TensorInfo& inputXInfo = GetTensorInfo(inputs[0]); in Execute() local
|
| H A D | BatchMatMulImpl.cpp | 16 const TensorInfo& inputXInfo, in BatchMatMul()
|
| /aosp_15_r20/external/armnn/src/armnnSerializer/test/ |
| H A D | SerializerTests.cpp | 100 const armnn::TensorInfo inputXInfo({2, 3, 4, 5}, armnn::DataType::Float32); variable
|