Lines Matching refs:dataLayout
20 const DataLayoutIndexed& dataLayout) in GetOffset() argument
22 if (dataLayout.GetDataLayout() == DataLayout::NHWC) in GetOffset()
24 … return ((b * shape[dataLayout.GetHeightIndex()] + h) * shape[dataLayout.GetWidthIndex()] + w) * in GetOffset()
25 shape[dataLayout.GetChannelsIndex()] + c; in GetOffset()
29 … return ((b * shape[dataLayout.GetChannelsIndex()] + c) * shape[dataLayout.GetHeightIndex()] + h) * in GetOffset()
30 shape[dataLayout.GetWidthIndex()] + w; in GetOffset()
40 DataLayoutIndexed dataLayout = params.m_DataLayout; in SpaceToBatchNd() local
45 const unsigned int channels = inputShape[dataLayout.GetChannelsIndex()]; in SpaceToBatchNd()
48 const unsigned int inputHeight = inputShape[dataLayout.GetHeightIndex()]; in SpaceToBatchNd()
49 const unsigned int inputWidth = inputShape[dataLayout.GetWidthIndex()]; in SpaceToBatchNd()
52 const unsigned int outputHeight = outputShape[dataLayout.GetHeightIndex()]; in SpaceToBatchNd()
53 const unsigned int outputWidth = outputShape[dataLayout.GetWidthIndex()]; in SpaceToBatchNd()
84 dataLayout); in SpaceToBatchNd()
99 dataLayout); in SpaceToBatchNd()
106 dataLayout); in SpaceToBatchNd()