Lines Matching refs:dataLayout
18 const DataLayoutIndexed& dataLayout) in GetOffset() argument
20 if (dataLayout.GetDataLayout() == armnn::DataLayout::NHWC) in GetOffset()
22 … return ((b * shape[dataLayout.GetHeightIndex()] + h) * shape[dataLayout.GetWidthIndex()] + w) * in GetOffset()
23 shape[dataLayout.GetChannelsIndex()] + c; in GetOffset()
27 … return ((b * shape[dataLayout.GetChannelsIndex()] + c) * shape[dataLayout.GetHeightIndex()] + h) * in GetOffset()
28 shape[dataLayout.GetWidthIndex()] + w; in GetOffset()
42 DataLayoutIndexed dataLayout = params.m_DataLayout; in SpaceToDepth() local
48 const unsigned int inputChannels = inputShape[dataLayout.GetChannelsIndex()]; in SpaceToDepth()
50 const unsigned int outputHeight = outputShape[dataLayout.GetHeightIndex()]; in SpaceToDepth()
51 const unsigned int outputWidth = outputShape[dataLayout.GetWidthIndex()]; in SpaceToDepth()
52 const unsigned int outputChannels = outputShape[dataLayout.GetChannelsIndex()]; in SpaceToDepth()
81 dataLayout); in SpaceToDepth()
88 dataLayout); in SpaceToDepth()