Lines Matching refs:srcShape
23 TransposeLoop(const armnn::TensorShape& srcShape, const armnn::PermutationVector& mappings) in TransposeLoop() argument
24 : m_SrcShape(srcShape) in TransposeLoop()
26 assert(srcShape.GetNumDimensions() == mappings.GetSize()); in TransposeLoop()
28 const size_type numDims = srcShape.GetNumDimensions(); in TransposeLoop()
38 srcStride *= srcShape[i]; in TransposeLoop()
39 dstStride *= srcShape[mappings[i]]; in TransposeLoop()
98 armnn::TensorShape TransposeTensorShape(const armnn::TensorShape& srcShape, const armnn::Permutatio… in TransposeTensorShape() argument
100 assert(srcShape.GetNumDimensions() == mappings.GetSize()); in TransposeTensorShape()
107 outDims[i] = srcShape[mappings[i]]; in TransposeTensorShape()
120 void Transpose(const armnn::TensorShape& srcShape, const armnn::PermutationVector& mappings, in Transpose() argument
123 TransposeLoop(srcShape, mappings).Unroll(src, dst, dataTypeSize); in Transpose()