diff options
Diffstat (limited to 'src/backends/reference/workloads/BatchToSpaceNd.cpp')
-rw-r--r-- | src/backends/reference/workloads/BatchToSpaceNd.cpp | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/src/backends/reference/workloads/BatchToSpaceNd.cpp b/src/backends/reference/workloads/BatchToSpaceNd.cpp index 5f64213b39..7efdb9b75c 100644 --- a/src/backends/reference/workloads/BatchToSpaceNd.cpp +++ b/src/backends/reference/workloads/BatchToSpaceNd.cpp @@ -37,8 +37,8 @@ void BatchToSpaceNd(const DataLayoutIndexed& dataLayout, const TensorInfo& outputTensorInfo, const std::vector<unsigned int>& blockShape, const std::vector<std::pair<unsigned int, unsigned int>>& cropsData, - const float* inputData, - float* outputData) + Decoder<float>& inputDecoder, + Encoder<float>& outputEncoder) { TensorShape inputShape = inputTensorInfo.GetShape(); @@ -90,7 +90,10 @@ void BatchToSpaceNd(const DataLayoutIndexed& dataLayout, { unsigned int outOffset = Offset(outputShape, outBatch, outH, outW, c, dataLayout); unsigned int inOffset = Offset(inputShape, inBatch, inH, inW, c, dataLayout); - outputData[outOffset] = inputData[inOffset]; + + outputEncoder[outOffset]; + inputDecoder[inOffset]; + outputEncoder.Set(inputDecoder.Get()); } } } |