From 41b164f36105c25456e9734aa2647db9e5de4308 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EA=B9=80=EC=88=98=EC=A7=84/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Engineer/=EC=82=BC=EC=84=B1=EC=A0=84?= =?utf8?q?=EC=9E=90?= Date: Mon, 10 Dec 2018 20:26:29 +0900 Subject: [PATCH] [neurun] Fix wrong parameter in nchw View.h (#3963) This commit fixes wrong parameter for 3 rank in nchw `View.h`. Signed-off-by: sjsujinkim --- runtimes/neurun/src/util/feature/nchw/View.h | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/runtimes/neurun/src/util/feature/nchw/View.h b/runtimes/neurun/src/util/feature/nchw/View.h index 9c3a9d9..104741d 100644 --- a/runtimes/neurun/src/util/feature/nchw/View.h +++ b/runtimes/neurun/src/util/feature/nchw/View.h @@ -51,7 +51,7 @@ public: public: T at(uint32_t ch, uint32_t row, uint32_t col) const override { - const auto offset = feature_index_to_byte_offset(ch, row, col); + const auto offset = feature_index_to_byte_offset(0, ch, row, col); T *ptr = reinterpret_cast(_tensor->buffer() + offset); @@ -69,7 +69,7 @@ public: public: T &at(uint32_t ch, uint32_t row, uint32_t col) { - const auto offset = feature_index_to_byte_offset(ch, row, col); + const auto offset = feature_index_to_byte_offset(0, ch, row, col); T *ptr = reinterpret_cast(_tensor->buffer() + offset); @@ -85,7 +85,7 @@ public: } private: - size_t feature_index_to_byte_offset(uint32_t batch, uint32_t ch = 0, uint32_t row = 0, + size_t feature_index_to_byte_offset(uint32_t batch = 0, uint32_t ch = 0, uint32_t row = 0, uint32_t col = 0) const { return _tensor->calcOffset( -- 2.7.4