Searched refs:m_DataLayoutY (Results 1 – 8 of 8) sorted by relevance
| /external/armnn/src/backends/cl/workloads/ |
| D | ClBatchMatMulWorkload.cpp | 38 …if (descriptor.m_DataLayoutX != armnn::DataLayout::NCHW || descriptor.m_DataLayoutY != armnn::Data… in ClBatchMatMulValidate()
50 …InputYInfo = armcomputetensorutils::BuildArmComputeTensorInfo(inputY, descriptor.m_DataLayoutY, 3); in ClBatchMatMulValidate()
51 …OutputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(output, descriptor.m_DataLayoutY, 3); in ClBatchMatMulValidate()
131 descriptor.m_Parameters.m_DataLayoutY != armnn::DataLayout::NCHW ) in ClBatchMatMulWorkload()
146 …fo()->set_data_layout(armcomputetensorutils::ConvertDataLayout(m_Data.m_Parameters.m_DataLayoutY)); in ClBatchMatMulWorkload()
|
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonBatchMatMulWorkload.cpp | 32 …if (descriptor.m_DataLayoutX != armnn::DataLayout::NCHW || descriptor.m_DataLayoutY != armnn::Data… in NeonBatchMatMulValidate()
38 …aclInputYInfo = armcomputetensorutils::BuildArmComputeTensorInfo(inputY, descriptor.m_DataLayoutY); in NeonBatchMatMulValidate()
111 descriptor.m_Parameters.m_DataLayoutY != armnn::DataLayout::NCHW ) in NeonBatchMatMulWorkload()
130 arm_compute::DataLayout aclDataLayoutY = ConvertDataLayout(m_Data.m_Parameters.m_DataLayoutY); in NeonBatchMatMulWorkload()
|
| /external/armnn/src/armnn/layers/ |
| D | BatchMatMulLayer.cpp | 49 auto permuteVec = BatchMatMulDescriptor::GetPermuteVec(m_Param.m_DataLayoutY, in InferOutputShapes()
66 m_Param.m_DataLayoutX : m_Param.m_DataLayoutY; in InferOutputShapes()
|
| /external/armnn/src/backends/reference/workloads/ |
| D | BatchMatMulImpl.cpp | 43 auto axesYToMul = BatchMatMulDescriptor::GetAxesToMul(params.m_DataLayoutY, in ApplyBatchMatMul()
120 auto permuteVec = BatchMatMulDescriptor::GetPermuteVec(params.m_DataLayoutY, in Transpose()
145 … const auto& dataLayout = (type == DataSlot::InputX) ? params.m_DataLayoutX : params.m_DataLayoutY; in Adjoint()
|
| /external/armnn/src/armnn/ |
| D | Descriptors.cpp | 465 GetAxesToMul(desc.m_DataLayoutY, tensorYShape) }; in GetAxesToMul()
473 GetAxesNotMul(desc.m_DataLayoutY, inputYShape) }; in GetAxesNotMul()
|
| /external/armnn/include/armnn/ |
| D | Descriptors.hpp | 1564 , m_DataLayoutY(dataLayoutY) in BatchMatMulDescriptor()
1574 m_DataLayoutY == rhs.m_DataLayoutY; in operator ==()
1589 DataLayout m_DataLayoutY; member
|
| /external/armnn/src/backends/backendsCommon/ |
| D | WorkloadData.cpp | 4154 m_Parameters.m_DataLayoutY, in Validate()
4159 auto axesToMul = BatchMatMulDescriptor::GetAxesToMul(m_Parameters.m_DataLayoutY, in Validate()
4192 switch(m_Parameters.m_DataLayoutY) in Validate()
4211 auto axesYToMul = BatchMatMulDescriptor::GetAxesToMul(m_Parameters.m_DataLayoutY, in Validate()
4225 DataLayout yLayout = m_Parameters.m_DataLayoutY; in Validate()
4274 auto axesYNotMul = BatchMatMulDescriptor::GetAxesNotMul(m_Parameters.m_DataLayoutY, in Validate()
|
| /external/armnn/src/armnnSerializer/ |
| D | Serializer.cpp | 237 … GetFlatBufferDataLayout(descriptor.m_DataLayoutY)); in SerializeBatchMatMulLayer()
|