tfinfo::v2::ShapeHint Class Reference

Class to represent axis and size of dims. User should enter axis and size of dim(s) when input tensor(s) contain(s) unknown dim(s). Such axis and size of dim(s) will be stored in ShapeHint. More...

#include <TensorSignature.h>

Public Member Functions
	ShapeHint ()=default
void	add (AxisHint axis, SizeHint size)
std::map< AxisHint, SizeHint >::const_iterator	cbegin () const
std::map< AxisHint, SizeHint >::const_iterator	cend () const
bool	empty ()
size_t	size ()

Detailed Description

Class to represent axis and size of dims. User should enter axis and size of dim(s) when input tensor(s) contain(s) unknown dim(s). Such axis and size of dim(s) will be stored in ShapeHint.

Definition at line 47 of file TensorSignature.h.

Constructor & Destructor Documentation

ShapeHint()

tfinfo::v2::ShapeHint::ShapeHint ( )

default

Member Function Documentation

add()

void tfinfo::v2::ShapeHint::add ( AxisHint  axis, SizeHint  size  )

inline

Definition at line 55 of file TensorSignature.h.

  {
    if (_dims.find(axis) != _dims.end())
      throw std::runtime_error("dim value already exists");
 
    _dims[axis] = size;
  }

References size().

cbegin()

std::map< AxisHint, SizeHint >::const_iterator tfinfo::v2::ShapeHint::cbegin ( ) const

inline

Definition at line 63 of file TensorSignature.h.

{ return _dims.cbegin(); }

cend()

std::map< AxisHint, SizeHint >::const_iterator tfinfo::v2::ShapeHint::cend ( ) const

inline

Definition at line 65 of file TensorSignature.h.

{ return _dims.cend(); }

empty()

bool tfinfo::v2::ShapeHint::empty ( )

inline

Definition at line 67 of file TensorSignature.h.

{ return _dims.size() == 0; }

size()

size_t tfinfo::v2::ShapeHint::size ( )

inline

Definition at line 69 of file TensorSignature.h.

{ return _dims.size(); }

Referenced by add().

---

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

• compiler/tfinfo-v2/include/tfinfo-v2/TensorSignature.h