nnfw::misc::kernel::IndexIterator Class Reference

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

#include <IndexIterator.h>

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

Detailed Description

Class to iterate Callable with Index of kernel.

Definition at line 38 of file IndexIterator.h.

Constructor & Destructor Documentation

IndexIterator()

nnfw::misc::kernel::IndexIterator::IndexIterator ( const Shape &  shape )

inline

Construct IndexIterator object with Shape of kernel.

Parameters

[in]

shape

Shape reference of feature

Definition at line 45 of file IndexIterator.h.

                                    : _shape{shape}
  {
    // DO NOTHING
  }

Member Function Documentation

iter()

template<typename Callable >

IndexIterator & nnfw::misc::kernel::IndexIterator::iter ( Callable  cb )

inline

Call a function iterated.

Parameters

[in]

cb

A callback function

Returns

Current IndexIterator object

Definition at line 56 of file IndexIterator.h.

  {
    for (int32_t nth = 0; nth < _shape.N; ++nth)
    {
      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(nth, ch, row, col);
          }
        }
      }
    }
 
    return (*this);
  }

References nnfw::misc::kernel::Shape::C, nnfw::misc::kernel::Shape::H, nnfw::misc::kernel::Shape::N, and nnfw::misc::kernel::Shape::W.

Referenced by nnfw::misc::kernel::operator<<().

---

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

• runtime/libs/misc/include/misc/kernel/IndexIterator.h