onert::exec::feature::IndexIterator Class Reference

Class to iterate Callable with Index of feature. More...

#include <IndexIterator.h>

Public Member Functions
	IndexIterator (const ir::FeatureShape &shape)
	Construct IndexIterator object with Shape of feature.
template<typename Callable >
IndexIterator &	iter (Callable cb)
	Call a function iterated.

Detailed Description

Class to iterate Callable with Index of feature.

Definition at line 33 of file IndexIterator.h.

Constructor & Destructor Documentation

IndexIterator()

onert::exec::feature::IndexIterator::IndexIterator ( const ir::FeatureShape &  shape )

inline

Construct IndexIterator object with Shape of feature.

Parameters

[in]

shape

Shape reference of feature

Definition at line 40 of file IndexIterator.h.

                                             : _shape{shape}
  {
    // DO NOTHING
  }

Member Function Documentation

iter()

template<typename Callable >

IndexIterator & onert::exec::feature::IndexIterator::iter ( Callable  cb )

inline

Call a function iterated.

Parameters

[in]

cb

A callback function

Returns

Current IndexIterator object

Definition at line 51 of file IndexIterator.h.

  {
    for (int32_t batch = 0; batch < _shape.N; ++batch)
    {
      for (int32_t ch = 0; ch < _shape.C; ++ch)
      {
        for (int32_t row = 0; row < _shape.H; ++row)
        {
          for (int32_t col = 0; col < _shape.W; ++col)
          {
            cb(batch, ch, row, col);
          }
        }
      }
    }
 
    return (*this);
  }

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

Referenced by onert::exec::feature::operator<<().

---

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

• runtime/onert/core/src/exec/feature/IndexIterator.h