runtime/onert/core/src/exec/feature/nchw/View.h

   1 /*
   2  * Copyright (c) 2018 Samsung Electronics Co., Ltd. All Rights Reserved
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #ifndef __ONERT_EXEC_FEATURE_NCHW_VIEW_H__
  18 #define __ONERT_EXEC_FEATURE_NCHW_VIEW_H__
  19
  20 #include "Reader.h"
  21
  22 #include "backend/ITensor.h"
  23 #include "ir/Shape.h"
  24 #include "util/logging.h"
  25
  26 #include <cassert>
  27
  28 namespace onert
  29 {
  30 namespace exec
  31 {
  32 namespace feature
  33 {
  34 namespace nchw
  35 {
  36
  37 template <typename T> class View final : public Reader<T>
  38 {
  39 public:
  40   // Construct for buffer of model inputs
  41   View(const ir::FeatureShape &shape, T *ptr, size_t len) : Reader<T>{shape, ptr, len}
  42   {
  43     // DO NOTHING
  44   }
  45
  46   // Construct for backend tensor
  47   View(::onert::backend::ITensor *tensor) : Reader<T>{tensor}
  48   {
  49     // DO NOTHING
  50   }
  51
  52 public:
  53   using Reader<T>::at;
  54   T &at(uint32_t batch, uint32_t ch, uint32_t row, uint32_t col)
  55   {
  56     return const_cast<T &>(Reader<T>::getRef(batch, ch, row, col));
  57   }
  58   T &at(uint32_t ch, uint32_t row, uint32_t col)
  59   {
  60     return const_cast<T &>(Reader<T>::getRef(0, ch, row, col));
  61   }
  62 };
  63
  64 } // namespace nchw
  65 } // namespace feature
  66 } // namespace exec
  67 } // namespace onert
  68
  69 #endif // __ONERT_EXEC_FEATURE_NCHW_VIEW_H__