Searched defs:inputYInfo (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
| H A D | BatchMatMulTestImpl.cpp | 29 const armnn::TensorInfo& inputYInfo, in BatchMatMulTestImpl()
78 armnn::TensorInfo inputYInfo({2,2}, ArmnnType, qScale, qOffset); in BatchMatMul2DSimpleTest() local
156 armnn::TensorInfo inputYInfo({1,2,2}, ArmnnType, qScale, qOffset); in BatchMatMul3DSimpleTest() local
234 armnn::TensorInfo inputYInfo({1,1,2,2}, ArmnnType, qScale, qOffset); in BatchMatMulNCHWSimpleTest() local
317 armnn::TensorInfo inputYInfo({1,2,2,1}, ArmnnType, qScale, qOffset); in BatchMatMulNHWCSimpleTest() local
395 armnn::TensorInfo inputYInfo({2,2,2}, ArmnnType, qScale, qOffset); in BatchMatMul3DBatchTest() local
482 armnn::TensorInfo inputYInfo({1,2,2}, ArmnnType, qScale, qOffset); in BatchMatMul3DBroadcastTest() local
566 armnn::TensorInfo inputYInfo({2,2}, ArmnnType, qScale, qOffset); in BatchMatMul3D2DBroadcastTest() local
655 armnn::TensorInfo inputYInfo({1,2,2,2}, ArmnnType, qScale, qOffset); in BatchMatMulNDHWCNHWCTest() local
742 armnn::TensorInfo inputYInfo({1,1}, ArmnnType, qScale, qOffset); in BatchMatMul2DTinyTest() local
[all …]
|
| /aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
| H A D | BatchMatMulImpl.hpp | 36 TensorInfo inputYInfo; member in armnn::BatchMatMul
|
| H A D | RefBatchMatMulWorkload.cpp | 35 const TensorInfo& inputYInfo = GetTensorInfo(inputs[1]); in Execute() local
|
| H A D | BatchMatMulImpl.cpp | 17 const TensorInfo& inputYInfo, in BatchMatMul()
|
| /aosp_15_r20/external/armnn/src/armnnSerializer/test/ |
| H A D | SerializerTests.cpp | 101 const armnn::TensorInfo inputYInfo({2, 4, 3, 5}, armnn::DataType::Float32); variable
|