onert::exec::feature::nchw::Reader< T > Class Template Reference

#include <Reader.h>

Collaboration diagram for onert::exec::feature::nchw::Reader< T >:

Public Types
using	Strides = ir::FeatureShape

Public Member Functions
	Reader (const ir::FeatureShape &shape, const Strides &strides, const T *ptr, size_t len)
	Reader (backend::ITensor *tensor)
T	at (uint32_t batch, uint32_t ch, uint32_t row, uint32_t col) const final
	Get the value used by four indexes.
T	at (uint32_t ch, uint32_t row, uint32_t col) const final
	Get the value used by three indexes.
Public Member Functions inherited from onert::exec::feature::Reader< T >
virtual	~Reader ()=default
	Destruct Reader object using default destructor.

Protected Member Functions
const T &	getRef (uint32_t batch, uint32_t ch, uint32_t row, uint32_t col) const

Detailed Description

template<typename T>
class onert::exec::feature::nchw::Reader< T >

Definition at line 36 of file Reader.h.

Member Typedef Documentation

Strides

template<typename T >

using onert::exec::feature::nchw::Reader< T >::Strides = ir::FeatureShape

Definition at line 39 of file Reader.h.

Constructor & Destructor Documentation

Reader() [1/2]

template<typename T >

onert::exec::feature::nchw::Reader< T >::Reader ( const ir::FeatureShape &  shape, const Strides &  strides, const T *  ptr, size_t  len  )

inline

Definition at line 41 of file Reader.h.

    : _shape{shape}, _strides{strides}, _ptr{reinterpret_cast<const uint8_t *>(ptr)}, _len{len}
  {
    assert(len == static_cast<size_t>(strides.N != 0   ? shape.N * strides.N
                                      : strides.C != 0 ? shape.C * strides.C
                                      : strides.H != 0 ? shape.H * strides.H
                                                       : shape.W * strides.W));
  }

References onert::ir::FeatureShape::C, onert::ir::FeatureShape::H, onert::ir::FeatureShape::N, and onert::ir::FeatureShape::W.

Reader() [2/2]

template<typename T >

onert::exec::feature::nchw::Reader< T >::Reader ( backend::ITensor *  tensor )

inline

Definition at line 51 of file Reader.h.

    : _ptr{tensor->buffer() + tensor->calcOffset({0, 0, 0, 0})}, _len{tensor->total_size()}
  {
    const auto start_offset = tensor->calcOffset({0, 0, 0, 0});
    auto shape = tensor->getShape();
    _strides.W = shape.dim(3) == 1 ? 0 : tensor->calcOffset({0, 0, 0, 1}) - start_offset;
    _strides.H = shape.dim(2) == 1 ? 0 : tensor->calcOffset({0, 0, 1, 0}) - start_offset;
    _strides.C = shape.dim(1) == 1 ? 0 : tensor->calcOffset({0, 1, 0, 0}) - start_offset;
    _strides.N = shape.dim(0) == 1 ? 0 : tensor->calcOffset({1, 0, 0, 0}) - start_offset;
 
    _shape.W = shape.dim(3);
    _shape.H = shape.dim(2);
    _shape.C = shape.dim(1);
    _shape.N = shape.dim(0);
  }

Member Function Documentation

at() [1/2]

template<typename T >

T onert::exec::feature::nchw::Reader< T >::at ( uint32_t  batch, uint32_t  ch, uint32_t  row, uint32_t  col  ) const

inlinefinalvirtual

Get the value used by four indexes.

Parameters

[in]	batch	The batch index
[in]	ch	The depth index
[in]	row	The height index
[in]	col	The width index

Returns

The value at the offset

Implements onert::exec::feature::Reader< T >.

Definition at line 68 of file Reader.h.

  {
    return getRef(batch, ch, row, col);
  }

at() [2/2]

template<typename T >

T onert::exec::feature::nchw::Reader< T >::at ( uint32_t  ch, uint32_t  row, uint32_t  col  ) const

inlinefinalvirtual

Get the value used by three indexes.

Parameters

[in]	ch	The depth index
[in]	row	The height index
[in]	col	The width index

Returns

The value at the offset

Implements onert::exec::feature::Reader< T >.

Definition at line 72 of file Reader.h.

{ return getRef(0, ch, row, col); }

getRef()

template<typename T >

const T & onert::exec::feature::nchw::Reader< T >::getRef ( uint32_t  batch, uint32_t  ch, uint32_t  row, uint32_t  col  ) const

inlineprotected

Definition at line 75 of file Reader.h.

  {
    const auto offset = feature_index_to_byte_offset(batch, ch, row, col);
 
    const T *ptr = reinterpret_cast<const T *>(_ptr + offset);
 
    return *ptr;
  }

References offset().

Referenced by onert::exec::feature::nchw::View< T >::at(), and onert::exec::feature::nchw::View< T >::at().

---

The documentation for this class was generated from the following file:

• runtime/onert/core/src/exec/feature/nchw/Reader.h