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/LogSoftMaxLayer.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/LogSoftMaxLayer.cc')
| -rw-r--r-- | runtime/onert/backend/cpu/ops/LogSoftMaxLayer.cc | 13 |
1 files changed, 6 insertions, 7 deletions
diff --git a/runtime/onert/backend/cpu/ops/LogSoftMaxLayer.cc b/runtime/onert/backend/cpu/ops/LogSoftMaxLayer.cc index 1d7ee6caa..a544dd970 100644 --- a/runtime/onert/backend/cpu/ops/LogSoftMaxLayer.cc +++ b/runtime/onert/backend/cpu/ops/LogSoftMaxLayer.cc @@ -49,9 +49,8 @@ void LogSoftMaxLayer::logsoftmaxFloat32() nnfw::cker::SoftmaxParams op_params; op_params.beta = _beta; op_params.axis = _axis; - nnfw::cker::LogSoftmax(op_params, getTensorShape(_input), - reinterpret_cast<const float *>(_input->buffer()), getTensorShape(_output), - reinterpret_cast<float *>(_output->buffer())); + nnfw::cker::LogSoftmax(op_params, getShape(_input), getBuffer<float>(_input), getShape(_output), + getBuffer<float>(_output)); } void LogSoftMaxLayer::logsoftmaxQuant8() @@ -60,11 +59,11 @@ void LogSoftMaxLayer::logsoftmaxQuant8() op_params.beta = _beta; op_params.axis = _axis; op_params.table = _table; - op_params.zero_point = _output->data_offset(); + op_params.zero_point = _output->data_zero_point(); op_params.scale = _output->data_scale(); - nnfw::cker::LogSoftmax(op_params, _input->data_scale(), getTensorShape(_input), - reinterpret_cast<const uint8_t *>(_input->buffer()), - getTensorShape(_output), reinterpret_cast<uint8_t *>(_output->buffer())); + nnfw::cker::LogSoftmax(op_params, _input->data_scale(), getShape(_input), + getBuffer<uint8_t>(_input), getShape(_output), + getBuffer<uint8_t>(_output)); } void LogSoftMaxLayer::configure(const IPortableTensor *input, const float beta, const int axis, |