20         unsigned int channels, const DataLayoutIndexed& dataLayout)  in Offset()  argument
22     if (dataLayout.GetDataLayout() == DataLayout::NHWC)  in Offset()
24 …return ((batch * shape[dataLayout.GetHeightIndex()] + height) * shape[dataLayout.GetWidthIndex()] …  in Offset()
25                shape[dataLayout.GetChannelsIndex()] + channels;  in Offset()
29         return ((batch * shape[dataLayout.GetChannelsIndex()] + channels) *  in Offset()
30                shape[dataLayout.GetHeightIndex()] + height) *  in Offset()
31                shape[dataLayout.GetWidthIndex()] + width;  in Offset()
35 void BatchToSpaceNd(const DataLayoutIndexed& dataLayout,  in BatchToSpaceNd()  argument
52     const unsigned int channels = inputShape[dataLayout.GetChannelsIndex()];  in BatchToSpaceNd()
55     const unsigned int outputHeight = outputShape[dataLayout.GetHeightIndex()];  in BatchToSpaceNd()
56     const unsigned int outputWidth = outputShape[dataLayout.GetWidthIndex()];  in BatchToSpaceNd()
73 … for (unsigned int inH = 0; inH < inputTensorInfo.GetShape()[dataLayout.GetHeightIndex()]; ++inH) {  in BatchToSpaceNd()
81 …  for (unsigned int inW = 0; inW < inputTensorInfo.GetShape()[dataLayout.GetWidthIndex()]; ++inW) {  in BatchToSpaceNd()
91 …                 unsigned int outOffset = Offset(outputShape, outBatch, outH, outW, c, dataLayout);  in BatchToSpaceNd()
92                     unsigned int inOffset = Offset(inputShape, inBatch, inH, inW, c, dataLayout);  in BatchToSpaceNd()