diff options
Diffstat (limited to 'runtime/onert/backend/cpu/ops/PowLayer.cc')
| -rw-r--r-- | runtime/onert/backend/cpu/ops/PowLayer.cc | 10 |
1 files changed, 4 insertions, 6 deletions
diff --git a/runtime/onert/backend/cpu/ops/PowLayer.cc b/runtime/onert/backend/cpu/ops/PowLayer.cc index 04a1af1e1..efd024dee 100644 --- a/runtime/onert/backend/cpu/ops/PowLayer.cc +++ b/runtime/onert/backend/cpu/ops/PowLayer.cc @@ -39,15 +39,13 @@ void PowLayer::powFloat32() if (!HaveSameShapes(_lhs, _rhs)) { nnfw::cker::BroadcastBinaryArithmeticOp<nnfw::cker::BinaryArithmeticOpType::POW>( - op_params, getTensorShape(_lhs), reinterpret_cast<const float *>(_lhs->buffer()), - getTensorShape(_rhs), reinterpret_cast<const float *>(_rhs->buffer()), - getTensorShape(_output), reinterpret_cast<float *>(_output->buffer())); + op_params, getShape(_lhs), getBuffer<float>(_lhs), getShape(_rhs), getBuffer<float>(_rhs), + getShape(_output), getBuffer<float>(_output)); return; } - nnfw::cker::powImpl(getTensorShape(_lhs), reinterpret_cast<const float *>(_lhs->buffer()), - getTensorShape(_rhs), reinterpret_cast<const float *>(_rhs->buffer()), - getTensorShape(_output), reinterpret_cast<float *>(_output->buffer())); + nnfw::cker::powImpl(getShape(_lhs), getBuffer<float>(_lhs), getShape(_rhs), + getBuffer<float>(_rhs), getShape(_output), getBuffer<float>(_output)); } void PowLayer::configure(const IPortableTensor *lhs, const IPortableTensor *rhs, |