Searched refs:srcShape (Results 1 – 7 of 7) sorted by relevance
| /external/armnn/src/armnnUtils/ |
| D | Transpose.cpp | 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
[all …]
|
| D | Permute.cpp | 98 armnn::TensorShape Permuted(const armnn::TensorShape& srcShape, in Permuted() argument
101 assert(srcShape.GetNumDimensions() == mappings.GetSize()); in Permuted()
108 outDims[mappings[i]] = srcShape[i]; in Permuted()
|
| /external/armnn/src/backends/backendsCommon/ |
| D | WorkloadUtils.hpp | 53 const TensorShape& srcShape = srcTensor->GetShape(); in CopyTensorContentsGeneric() local
54 const auto srcSize = srcTensor->GetStrides()[0] * srcShape[0]; in CopyTensorContentsGeneric()
66 AssignValues(srcShape.GetNumDimensions(), in CopyTensorContentsGeneric()
68 srcShape, in CopyTensorContentsGeneric()
|
| /external/armnn/include/armnnUtils/ |
| D | Permute.hpp | 16 armnn::TensorShape Permuted(const armnn::TensorShape& srcShape,
|
| D | Transpose.hpp | 15 armnn::TensorShape TransposeTensorShape(const armnn::TensorShape& srcShape, const armnn::Permutatio…
|
| /external/tensorflow/tensorflow/lite/java/src/main/java/org/tensorflow/lite/ |
| D | TensorImpl.java | 423 int[] srcShape = computeShapeOf(src); in throwIfSrcShapeIsIncompatible() local
424 if (!Arrays.equals(srcShape, shapeCopy)) { in throwIfSrcShapeIsIncompatible()
429 name(), Arrays.toString(shapeCopy), Arrays.toString(srcShape))); in throwIfSrcShapeIsIncompatible()
|
| /external/pytorch/aten/src/ATen/native/mps/operations/ |
| D | Copy.mm | 47 MPSShape* srcShape = getMPSShape(src);
71 shape:srcShape
|