diff options
Diffstat (limited to 'runtime/onert/backend/cpu/ops/RangeLayer.cc')
| -rw-r--r-- | runtime/onert/backend/cpu/ops/RangeLayer.cc | 12 |
1 files changed, 4 insertions, 8 deletions
diff --git a/runtime/onert/backend/cpu/ops/RangeLayer.cc b/runtime/onert/backend/cpu/ops/RangeLayer.cc index f00101fa8..a41b31b3f 100644 --- a/runtime/onert/backend/cpu/ops/RangeLayer.cc +++ b/runtime/onert/backend/cpu/ops/RangeLayer.cc @@ -47,16 +47,12 @@ void RangeLayer::run() switch (_output->data_type()) { case OperandType::FLOAT32: - nnfw::cker::Range<float>(reinterpret_cast<float *>(_start->buffer()), - reinterpret_cast<float *>(_limit->buffer()), - reinterpret_cast<float *>(_delta->buffer()), - reinterpret_cast<float *>(_output->buffer())); + nnfw::cker::Range<float>(getBuffer<float>(_start), getBuffer<float>(_limit), + getBuffer<float>(_delta), getBuffer<float>(_output)); break; case OperandType::INT32: - nnfw::cker::Range<int32_t>(reinterpret_cast<int32_t *>(_start->buffer()), - reinterpret_cast<int32_t *>(_limit->buffer()), - reinterpret_cast<int32_t *>(_delta->buffer()), - reinterpret_cast<int32_t *>(_output->buffer())); + nnfw::cker::Range<int32_t>(getBuffer<int32_t>(_start), getBuffer<int32_t>(_limit), + getBuffer<int32_t>(_delta), getBuffer<int32_t>(_output)); break; default: throw std::runtime_error{"Range: unsupported data type"}; |