Home
last modified time | relevance | path

Searched refs:GetHeightIndex (Results 1 – 25 of 38) sorted by relevance

12

/external/armnn/src/backends/reference/workloads/
DBatchToSpaceNd.cpp24 …return ((batch * shape[dataLayout.GetHeightIndex()] + height) * shape[dataLayout.GetWidthIndex()] … in Offset()
30 shape[dataLayout.GetHeightIndex()] + height) * in Offset()
55 const unsigned int outputHeight = outputShape[dataLayout.GetHeightIndex()]; in BatchToSpaceNd()
73 … for (unsigned int inH = 0; inH < inputTensorInfo.GetShape()[dataLayout.GetHeightIndex()]; ++inH) { in BatchToSpaceNd()
DSpaceToBatchNd.cpp24 … return ((b * shape[dataLayout.GetHeightIndex()] + h) * shape[dataLayout.GetWidthIndex()] + w) * in GetOffset()
29 … return ((b * shape[dataLayout.GetChannelsIndex()] + c) * shape[dataLayout.GetHeightIndex()] + h) * in GetOffset()
48 const unsigned int inputHeight = inputShape[dataLayout.GetHeightIndex()]; in SpaceToBatchNd()
52 const unsigned int outputHeight = outputShape[dataLayout.GetHeightIndex()]; in SpaceToBatchNd()
DSpaceToDepth.cpp22 … return ((b * shape[dataLayout.GetHeightIndex()] + h) * shape[dataLayout.GetWidthIndex()] + w) * in GetOffset()
27 … return ((b * shape[dataLayout.GetChannelsIndex()] + c) * shape[dataLayout.GetHeightIndex()] + h) * in GetOffset()
50 const unsigned int outputHeight = outputShape[dataLayout.GetHeightIndex()]; in SpaceToDepth()
DResize.cpp84 const unsigned int inputHeight = inputInfo.GetShape()[dataLayout.GetHeightIndex()]; in Resize()
86 const unsigned int outputHeight = outputInfo.GetShape()[dataLayout.GetHeightIndex()]; in Resize()
DDepthToSpace.cpp32 const unsigned int inHeight = inputShape[dataLayoutIndexed.GetHeightIndex()]; in DepthToSpace()
DBatchNormImpl.cpp32 unsigned int inputHeight = inputShape[dataLayout.GetHeightIndex()]; in BatchNormImpl()
DInstanceNorm.cpp28 unsigned int inputHeight = inputShape[dataLayout.GetHeightIndex()]; in InstanceNorm()
DRefL2NormalizationWorkload.cpp63 const int heightIdx = armnn::numeric_cast<int>(dataLayout.GetHeightIndex()); in Execute()
DTransposeConvolution2d.cpp30 const unsigned int heightIndex = dataLayoutIndexed.GetHeightIndex(); in TransposeConvolution2dImpl()
DConv3dImpl.cpp37 const unsigned int heightIndex = dataLayoutIndexed.GetHeightIndex(); in Convolve3d()
DRefNormalizationWorkload.cpp105 const unsigned int rows = tensorShape[dataLayoutIndexed.GetHeightIndex()]; in NormalizeAcrossUingLbr()
DConvImpl.cpp95 const unsigned int heightIndex = dataLayoutIndexed.GetHeightIndex(); in Convolve()
/external/armnn/src/armnn/layers/
DConvolution2dLayer.cpp40 unsigned int filterHeight = filterShape[dataLayoutIndex.GetHeightIndex()]; in SerializeLayerParameters()
79 unsigned int inHeight = inputShape[dataLayoutIndex.GetHeightIndex()]; in InferOutputShapes()
87 unsigned int filterHeight = filterShape[dataLayoutIndex.GetHeightIndex()]; in InferOutputShapes()
DTransposeConvolution2dLayer.cpp71 const unsigned int hInput = inputShape[dataLayoutIndex.GetHeightIndex()]; in InferOutputShapes()
74 const unsigned int hKernel = kernelShape[dataLayoutIndex.GetHeightIndex()]; in InferOutputShapes()
DSpaceToDepthLayer.cpp52 unsigned int hIndex = dimensionIndices.GetHeightIndex(); in InferOutputShapes()
DDepthToSpaceLayer.cpp50 unsigned int hIndex = dimensionIndices.GetHeightIndex(); in InferOutputShapes()
DSpaceToBatchNdLayer.cpp56 unsigned int heightIndex = dimensionIndices.GetHeightIndex(); in InferOutputShapes()
DBatchToSpaceNdLayer.cpp77 unsigned int heightIndex = dimensionIndices.GetHeightIndex(); in InferOutputShapes()
DPooling2dLayer.cpp50 unsigned int inHeight = inputShape[dimensionIndices.GetHeightIndex()]; in InferOutputShapes()
DPooling3dLayer.cpp50 unsigned int inHeight = inputShape[dimensionIndices.GetHeightIndex()]; in InferOutputShapes()
DDepthwiseConvolution2dLayer.cpp79 unsigned int inputHeight = inputShape[dataLayoutIndex.GetHeightIndex()]; in InferOutputShapes()
DConvolution3dLayer.cpp77 unsigned int inHeight = inputShape[dataLayoutIndex.GetHeightIndex()]; in InferOutputShapes()
/external/armnn/src/armnn/optimizations/
DFoldPadIntoLayer2d.hpp109 layerDescriptor.m_PadTop += padList[layout.GetHeightIndex()].first; in TryFoldPadIntoLayer2d()
110 layerDescriptor.m_PadBottom += padList[layout.GetHeightIndex()].second; in TryFoldPadIntoLayer2d()
/external/armnn/include/armnnUtils/
DDataLayoutIndexed.hpp24 unsigned int GetHeightIndex() const { return m_HeightIndex; } in GetHeightIndex() function in armnnUtils::DataLayoutIndexed
/external/armnn/src/backends/backendsCommon/test/
DSpaceToDepthEndToEndTestImpl.cpp40 if (inputShape[dimensionIndices.GetHeightIndex()] % blockSize!=0 in CreateSpaceToDepthNetwork()

12