Searched refs:dstShape (Results 1 – 7 of 7) sorted by relevance
| /external/armnn/src/armnnUtils/ |
| D | Permute.cpp | 23 PermuteLoop(const armnn::TensorShape& dstShape, const armnn::PermutationVector& mappings) in PermuteLoop() argument
24 : m_DstShape(dstShape) in PermuteLoop()
26 assert(dstShape.GetNumDimensions() == mappings.GetSize()); in PermuteLoop()
28 const size_type numDims = dstShape.GetNumDimensions(); in PermuteLoop()
38 srcStride *= dstShape[mappings[i]]; in PermuteLoop()
39 dstStride *= dstShape[i]; in PermuteLoop()
131 void Permute(const armnn::TensorShape& dstShape, const armnn::PermutationVector& mappings, in Permute() argument
134 PermuteLoop(dstShape, mappings).Unroll(src, dst, dataTypeSize); in Permute()
|
| /external/armnn/src/backends/backendsCommon/ |
| D | WorkloadUtils.hpp | 57 const TensorShape& dstShape = dstTensor->GetShape(); in CopyTensorContentsGeneric() local
58 const auto dstSize = dstTensor->GetStrides()[0] * dstShape[0]; in CopyTensorContentsGeneric()
94 AssignValues(dstShape.GetNumDimensions(), in CopyTensorContentsGeneric()
96 dstShape, in CopyTensorContentsGeneric()
|
| /external/armnn/tests/ |
| D | ImagePreprocessor.cpp | 45 armnn::TensorShape dstShape({m_BatchSize, 3, m_Height, m_Width}); in GetLabelAndResizedImageAsFloat() local
47 armnnUtils::Permute(dstShape, NHWCToArmNN, result.data(), tempImage.data(), sizeof(float)); in GetLabelAndResizedImageAsFloat()
|
| /external/armnn/tests/ImageTensorGenerator/ |
| D | ImageTensorGenerator.hpp | 118 armnn::TensorShape dstShape({ batchSize, 3, newHeight, newWidth }); in PrepareImageTensor() local
120 … armnnUtils::Permute(dstShape, NHWCToArmNN, imageData.data(), tempImage.data(), sizeof(float)); in PrepareImageTensor()
|
| /external/armnn/include/armnnUtils/ |
| D | Transpose.hpp | 19 void Transpose(const armnn::TensorShape& dstShape, const armnn::PermutationVector& mappings,
|
| D | Permute.hpp | 22 void Permute(const armnn::TensorShape& dstShape, const armnn::PermutationVector& mappings,
|
| /external/tensorflow/tensorflow/lite/java/src/main/java/org/tensorflow/lite/ |
| D | TensorImpl.java | 451 int[] dstShape = computeShapeOf(dst); in throwIfDstShapeIsIncompatible() local
452 if (!Arrays.equals(dstShape, shapeCopy)) { in throwIfDstShapeIsIncompatible()
457 name(), Arrays.toString(shapeCopy), Arrays.toString(dstShape))); in throwIfDstShapeIsIncompatible()
|