diff options
Diffstat (limited to 'runtime/onert/backend/cpu/ops/UnpackLayer.cc')
| -rw-r--r-- | runtime/onert/backend/cpu/ops/UnpackLayer.cc | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/runtime/onert/backend/cpu/ops/UnpackLayer.cc b/runtime/onert/backend/cpu/ops/UnpackLayer.cc index 428b38588..f18fb9483 100644 --- a/runtime/onert/backend/cpu/ops/UnpackLayer.cc +++ b/runtime/onert/backend/cpu/ops/UnpackLayer.cc @@ -47,7 +47,7 @@ template <typename T> void UnpackLayer::unpackImpl() for (int32_t i = 0; i < _num_output; i++) { - outputDims.push_back(getTensorShape(_outputs[i])); + outputDims.push_back(getShape(_outputs[i])); outputDimsPtr.push_back(&outputDims[i]); } @@ -55,11 +55,11 @@ template <typename T> void UnpackLayer::unpackImpl() for (const auto output : _outputs) { - outputPtrs.emplace_back(reinterpret_cast<T *>(output->buffer())); + outputPtrs.emplace_back(getBuffer<T>(output)); } - nnfw::cker::Unpack<T>(op_params, getTensorShape(_input), reinterpret_cast<T *>(_input->buffer()), - getTensorShape(_outputs[0]), outputPtrs.data()); + nnfw::cker::Unpack<T>(op_params, getShape(_input), getBuffer<T>(_input), getShape(_outputs[0]), + outputPtrs.data()); } void UnpackLayer::configure(const IPortableTensor *input, uint32_t axis, int32_t num, |