Home
last modified time | relevance | path

Searched refs:dataLayoutIndexed (Results 1 – 11 of 11) sorted by relevance

/external/armnn/src/backends/reference/workloads/
DConv3dImpl.cpp34 const armnnUtils::DataLayoutIndexed dataLayoutIndexed(dataLayout); in Convolve3d() local
36 const unsigned int channelsIndex = dataLayoutIndexed.GetChannelsIndex(); in Convolve3d()
37 const unsigned int heightIndex = dataLayoutIndexed.GetHeightIndex(); in Convolve3d()
38 const unsigned int widthIndex = dataLayoutIndexed.GetWidthIndex(); in Convolve3d()
39 const unsigned int depthIndex = dataLayoutIndexed.GetDepthIndex(); in Convolve3d()
116 … if (dataLayoutIndexed.GetDataLayout() == DataLayout::NDHWC) in Convolve3d()
151 if (dataLayoutIndexed.GetDataLayout() == DataLayout::NDHWC) in Convolve3d()
DDepthToSpace.cpp30 armnnUtils::DataLayoutIndexed dataLayoutIndexed(descriptor.m_DataLayout); in DepthToSpace() local
31 const unsigned int inDepth = inputShape[dataLayoutIndexed.GetChannelsIndex()]; in DepthToSpace()
32 const unsigned int inHeight = inputShape[dataLayoutIndexed.GetHeightIndex()]; in DepthToSpace()
33 const unsigned int inWidth = inputShape[dataLayoutIndexed.GetWidthIndex()]; in DepthToSpace()
DConvImpl.cpp92 const armnnUtils::DataLayoutIndexed dataLayoutIndexed(dataLayout); in Convolve() local
94 const unsigned int channelsIndex = dataLayoutIndexed.GetChannelsIndex(); in Convolve()
95 const unsigned int heightIndex = dataLayoutIndexed.GetHeightIndex(); in Convolve()
96 const unsigned int widthIndex = dataLayoutIndexed.GetWidthIndex(); in Convolve()
154 if (dataLayoutIndexed.GetDataLayout() == DataLayout::NHWC) in Convolve()
187 if (dataLayoutIndexed.GetDataLayout() == DataLayout::NHWC) in Convolve()
215 if (dataLayoutIndexed.GetDataLayout() == DataLayout::NHWC) in Convolve()
DTransposeConvolution2d.cpp28 const DataLayoutIndexed dataLayoutIndexed(descriptor.m_DataLayout); in TransposeConvolution2dImpl() local
29 const unsigned int channelsIndex = dataLayoutIndexed.GetChannelsIndex(); in TransposeConvolution2dImpl()
30 const unsigned int heightIndex = dataLayoutIndexed.GetHeightIndex(); in TransposeConvolution2dImpl()
31 const unsigned int widthIndex = dataLayoutIndexed.GetWidthIndex(); in TransposeConvolution2dImpl()
146 … dataLayoutIndexed.GetIndex(outputShape, batch, dOutput, yOutput, xOutput); in TransposeConvolution2dImpl()
DRefNormalizationWorkload.cpp101 DataLayoutIndexed dataLayoutIndexed(dataLayout); in NormalizeAcrossUingLbr() local
104 const unsigned int depth = tensorShape[dataLayoutIndexed.GetChannelsIndex()]; in NormalizeAcrossUingLbr()
105 const unsigned int rows = tensorShape[dataLayoutIndexed.GetHeightIndex()]; in NormalizeAcrossUingLbr()
106 const unsigned int cols = tensorShape[dataLayoutIndexed.GetWidthIndex()]; in NormalizeAcrossUingLbr()
128 unsigned inputIndex = dataLayoutIndexed.GetIndex(tensorShape, in NormalizeAcrossUingLbr()
143 unsigned index = dataLayoutIndexed.GetIndex(tensorShape, n, c, h, w); in NormalizeAcrossUingLbr()
/external/android-nn-driver/
DConversionUtils_1_2.hpp456 armnnUtils::DataLayoutIndexed dataLayoutIndexed(desc.m_DataLayout); in ConvertConv2d_1_2() local
457 unsigned int widthIndex = dataLayoutIndexed.GetWidthIndex(); in ConvertConv2d_1_2()
458 unsigned int heightIndex = dataLayoutIndexed.GetHeightIndex(); in ConvertConv2d_1_2()
589 armnnUtils::DataLayoutIndexed dataLayoutIndexed(desc.m_DataLayout); in ConvertDepthwiseConv2d_1_2() local
590 unsigned int widthIndex = dataLayoutIndexed.GetWidthIndex(); in ConvertDepthwiseConv2d_1_2()
591 unsigned int heightIndex = dataLayoutIndexed.GetHeightIndex(); in ConvertDepthwiseConv2d_1_2()
1038 armnnUtils::DataLayoutIndexed dataLayoutIndexed(dataLayout); in ConvertGroupedConv2d() local
1039 const unsigned int channelsIndex = dataLayoutIndexed.GetChannelsIndex(); in ConvertGroupedConv2d()
1040 const unsigned int heightIndex = dataLayoutIndexed.GetHeightIndex(); in ConvertGroupedConv2d()
1041 const unsigned int widthIndex = dataLayoutIndexed.GetWidthIndex(); in ConvertGroupedConv2d()
[all …]
/external/armnn/shim/sl/canonical/
DConverter.cpp1101 armnnUtils::DataLayoutIndexed dataLayoutIndexed(desc.m_DataLayout); in ConvertConv2d() local
1102 unsigned int widthIndex = dataLayoutIndexed.GetWidthIndex(); in ConvertConv2d()
1103 unsigned int heightIndex = dataLayoutIndexed.GetHeightIndex(); in ConvertConv2d()
1321 armnnUtils::DataLayoutIndexed dataLayoutIndexed(desc.m_DataLayout); in ConvertDepthwiseConv2d() local
1322 unsigned int widthIndex = dataLayoutIndexed.GetWidthIndex(); in ConvertDepthwiseConv2d()
1323 unsigned int heightIndex = dataLayoutIndexed.GetHeightIndex(); in ConvertDepthwiseConv2d()
2188 armnnUtils::DataLayoutIndexed dataLayoutIndexed(dataLayout); in ConvertGroupedConv2d() local
2189 const unsigned int channelsIndex = dataLayoutIndexed.GetChannelsIndex(); in ConvertGroupedConv2d()
2190 const unsigned int heightIndex = dataLayoutIndexed.GetHeightIndex(); in ConvertGroupedConv2d()
2191 const unsigned int widthIndex = dataLayoutIndexed.GetWidthIndex(); in ConvertGroupedConv2d()
[all …]
/external/armnn/src/backends/backendsCommon/test/layerTests/
DBatchNormalizationTestImpl.cpp45 armnnUtils::DataLayoutIndexed dataLayoutIndexed(dataLayout); in BatchNormTestImpl() local
47 armnn::TensorInfo tensorInfo({ inputOutputTensorShape[dataLayoutIndexed.GetChannelsIndex()] }, in BatchNormTestImpl()
DResizeTestImpl.cpp118 armnnUtils::DataLayoutIndexed dataLayoutIndexed(params.m_DataLayout); in ResizeTestImpl() local
119 … descriptor.m_Parameters.m_TargetWidth = params.m_OutputShape[dataLayoutIndexed.GetWidthIndex()]; in ResizeTestImpl()
120 … descriptor.m_Parameters.m_TargetHeight = params.m_OutputShape[dataLayoutIndexed.GetHeightIndex()]; in ResizeTestImpl()
DConv3dTestImpl.cpp36 const armnnUtils::DataLayoutIndexed dataLayoutIndexed(layout); in GetBiasData() local
37 … const unsigned int outputChannels = outputInfo.GetShape()[dataLayoutIndexed.GetChannelsIndex()]; in GetBiasData()
DConv2dTestImpl.cpp117 const armnnUtils::DataLayoutIndexed dataLayoutIndexed(layout); in GetBias() local
118 const unsigned int channelsIndex = dataLayoutIndexed.GetChannelsIndex(); in GetBias()