diff options
author | Éanna Ó Catháin <eanna.ocathain@arm.com> | 2018-11-12 17:14:43 +0000 |
---|---|---|
committer | Saoirse Stewart Arm <saoirse.stewart@arm.com> | 2018-11-12 21:59:48 +0000 |
commit | 95807cef855738ca481ace30f32ed9f245a098dd (patch) | |
tree | 7df7181d6dc19f3db3054614076478af4f417e8a /src/backends/reference/workloads/BatchToSpaceNd.hpp | |
parent | 111b5d94d7e854c21377f8d2c0b4234317a903f6 (diff) | |
download | armnn-95807cef855738ca481ace30f32ed9f245a098dd.tar.gz |
Tidying up multiple issues
* Fixed error in InferOutputShape implementation
* Added better error checking to the BatchToSpace implementation.
* Added defaults to the batchToSpace descriptors.
* Changed crops to be a vector of pairs to align with the SpaceToBatch implementation
Change-Id: Ib1c16d871f0898a1caeb6629c1fee6380a773e14
Diffstat (limited to 'src/backends/reference/workloads/BatchToSpaceNd.hpp')
-rw-r--r-- | src/backends/reference/workloads/BatchToSpaceNd.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/backends/reference/workloads/BatchToSpaceNd.hpp b/src/backends/reference/workloads/BatchToSpaceNd.hpp index 7923ceadd0..091d092777 100644 --- a/src/backends/reference/workloads/BatchToSpaceNd.hpp +++ b/src/backends/reference/workloads/BatchToSpaceNd.hpp @@ -16,7 +16,7 @@ void BatchToSpaceNd(const DataLayoutIndexed& dataLayout, const TensorInfo& inputTensorInfo, const TensorInfo& outputTensorInfo, const std::vector<unsigned int>& blockShape, - const std::vector<std::vector<unsigned int>>& cropsData, + const std::vector<std::pair<unsigned int, unsigned int>>& cropsData, const float* inputData, float* outputData); } // namespace armnn
\ No newline at end of file |