diff options
author | Chunseok Lee <chunseok.lee@samsung.com> | 2021-04-20 18:01:41 +0900 |
---|---|---|
committer | Chunseok Lee <chunseok.lee@samsung.com> | 2021-04-20 18:01:41 +0900 |
commit | 589bb1db6db6784efe21b3fbbfbfdb79aaa5f14e (patch) | |
tree | 47a2b23ce4220e3a4150c8b12ed941555272fb0c /runtime/onert/backend/cpu/ops/SpaceToDepthLayer.cc | |
parent | 62529acabbafce7730601ed01d5709d7bc0d378a (diff) | |
download | nnfw-589bb1db6db6784efe21b3fbbfbfdb79aaa5f14e.tar.gz nnfw-589bb1db6db6784efe21b3fbbfbfdb79aaa5f14e.tar.bz2 nnfw-589bb1db6db6784efe21b3fbbfbfdb79aaa5f14e.zip |
Imported Upstream version 1.15.0upstream/1.15.0submit/tizen/20210427.093759submit/tizen/20210423.055448submit/tizen/20210422.015846submit/tizen/20210421.062230accepted/tizen/unified/20210428.040443
Diffstat (limited to 'runtime/onert/backend/cpu/ops/SpaceToDepthLayer.cc')
-rw-r--r-- | runtime/onert/backend/cpu/ops/SpaceToDepthLayer.cc | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/runtime/onert/backend/cpu/ops/SpaceToDepthLayer.cc b/runtime/onert/backend/cpu/ops/SpaceToDepthLayer.cc index a0869aed8..8271daf42 100644 --- a/runtime/onert/backend/cpu/ops/SpaceToDepthLayer.cc +++ b/runtime/onert/backend/cpu/ops/SpaceToDepthLayer.cc @@ -39,9 +39,8 @@ template <typename T> void SpaceToDepthLayer::spaceToDepth() nnfw::cker::SpaceToDepthParams params; params.block_size = _block_size; - nnfw::cker::SpaceToDepth(params, getTensorShape(_input), - reinterpret_cast<const T *>(_input->buffer()), getTensorShape(_output), - reinterpret_cast<T *>(_output->buffer())); + nnfw::cker::SpaceToDepth(params, getShape(_input), getBuffer<T>(_input), getShape(_output), + getBuffer<T>(_output)); } void SpaceToDepthLayer::configure(const IPortableTensor *input, const int32_t block_size, |