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/StatelessRandomUniformLayer.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/StatelessRandomUniformLayer.cc')
-rw-r--r-- | runtime/onert/backend/cpu/ops/StatelessRandomUniformLayer.cc | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/runtime/onert/backend/cpu/ops/StatelessRandomUniformLayer.cc b/runtime/onert/backend/cpu/ops/StatelessRandomUniformLayer.cc index b8dfcb4b5..587582e8f 100644 --- a/runtime/onert/backend/cpu/ops/StatelessRandomUniformLayer.cc +++ b/runtime/onert/backend/cpu/ops/StatelessRandomUniformLayer.cc @@ -28,7 +28,7 @@ namespace ops { StatelessRandomUniformLayer::StatelessRandomUniformLayer() - : _shape(nullptr), _seed(nullptr), _output(nullptr) + : _shape(nullptr), _seed(nullptr), _output(nullptr) { // DO NOTHING } @@ -43,10 +43,9 @@ void StatelessRandomUniformLayer::configure(const IPortableTensor *shape, void StatelessRandomUniformLayer::StatelessRandomUniformFloat32() { - nnfw::cker::StatelessRandomUniform( - getTensorShape(_shape), reinterpret_cast<const int *>(_shape->buffer()), - getTensorShape(_seed), reinterpret_cast<const int *>(_seed->buffer()), - getTensorShape(_output), reinterpret_cast<float *>(_output->buffer())); + nnfw::cker::StatelessRandomUniform(getShape(_shape), getBuffer<int32_t>(_shape), getShape(_seed), + getBuffer<int32_t>(_seed), getShape(_output), + getBuffer<float>(_output)); } void StatelessRandomUniformLayer::run() |