moco::TensorSliceEnumerator Class Reference

#include <TensorSliceEnumerator.h>

Public Member Functions
	TensorSliceEnumerator ()=default
void	shape (loco::TensorShape &s)
void	begin (u32v_t &b)
void	end (u32v_t &e)
void	start (void)
bool	valid (void)
void	advance (void)
uint32_t	cursor (uint32_t rank) const
const u32v_t	cursor (void) const

Detailed Description

Definition at line 29 of file TensorSliceEnumerator.h.

Constructor & Destructor Documentation

TensorSliceEnumerator()

moco::TensorSliceEnumerator::TensorSliceEnumerator ( )

default

Member Function Documentation

advance()

void moco::TensorSliceEnumerator::advance

(

void

)

Definition at line 72 of file TensorSliceEnumerator.cpp.

{
  uint32_t r = _rank_in - 1;
  increment(r);
}

begin()

void moco::TensorSliceEnumerator::begin ( u32v_t &  b )

inline

Definition at line 36 of file TensorSliceEnumerator.h.

{ _begin = b; }

cursor() [1/2]

uint32_t moco::TensorSliceEnumerator::cursor

(

uint32_t

rank

)

const

Definition at line 78 of file TensorSliceEnumerator.cpp.

{
  assert(rank < _rank_in);
  return _cursor.at(rank);
}

cursor() [2/2]

const u32v_t moco::TensorSliceEnumerator::cursor ( void  ) const

inline

Definition at line 45 of file TensorSliceEnumerator.h.

{ return _cursor; }

end()

void moco::TensorSliceEnumerator::end ( u32v_t &  e )

inline

Definition at line 37 of file TensorSliceEnumerator.h.

{ _end = e; }

shape()

void moco::TensorSliceEnumerator::shape

(

loco::TensorShape &

s

)

Definition at line 24 of file TensorSliceEnumerator.cpp.

{
  _shape_in = s;
  _rank_in = _shape_in.rank();
}

References loco::TensorShape::rank().

Referenced by RandomDataGenerator.RandomDataGenerator::_gen_float32(), RandomDataGenerator.RandomDataGenerator::_gen_int16(), and RandomDataGenerator.RandomDataGenerator::_gen_uint8().

start()

void moco::TensorSliceEnumerator::start

(

void

)

Definition at line 48 of file TensorSliceEnumerator.cpp.

{
  auto rank = _rank_in;
 
  _cursor.resize(rank);
  _boundary.resize(rank);
  for (uint32_t r = 0; r < rank; ++r)
  {
    _cursor.at(r) = _begin[r];
    _boundary.at(r) = _end[r];
  }
}

valid()

bool moco::TensorSliceEnumerator::valid

(

void

)

Definition at line 61 of file TensorSliceEnumerator.cpp.

{
  auto rank = _rank_in;
  for (uint32_t r = 0; r < rank; ++r)
  {
    if (_cursor.at(r) >= _boundary.at(r))
      return false;
  }
  return true;
}

---

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

• compiler/moco/pass/src/TensorSliceEnumerator.h
• compiler/moco/pass/src/TensorSliceEnumerator.cpp