ONE - On-device Neural Engine
|
Class that provides index of tensor for training. More...
#include <Index.h>
Public Member Functions | |
TrainingIndex () | |
Construct TrainingOperationIndex object. | |
TrainingIndex (const T &index, bool is_forward) | |
Construct TrainingOperationIndex object. | |
const T & | index () const |
Get index. | |
bool | is_forward () const |
Get whether the tensor is forward tensor or not. | |
bool | valid () const |
Check if the index is valid or not. | |
bool | operator== (const TrainingIndex &other) const |
operator overloading function for == | |
bool | operator!= (const TrainingIndex &other) const |
operator overloading function for != | |
bool | operator< (const TrainingIndex &other) const |
operator overloading function for < | |
Class that provides index of tensor for training.
T | Type of index |
|
inline |
|
inline |
Construct TrainingOperationIndex object.
T | Type of index |
index | The operation index |
is_forward | Whether the tensor is forward tensor or not |
Definition at line 56 of file Index.h.
|
inline |
Get index.
Definition at line 67 of file Index.h.
Referenced by onert::ir::train::operator<<(), onert::ir::train::operator<<(), and onert::ir::train::TrainingIndex< T >::operator==().
|
inline |
Get whether the tensor is forward tensor or not.
Definition at line 73 of file Index.h.
Referenced by onert::ir::train::TrainingIndex< T >::operator==().
|
inline |
|
inline |
|
inline |
operator overloading function for ==
Definition at line 89 of file Index.h.
References onert::ir::train::TrainingIndex< T >::index(), and onert::ir::train::TrainingIndex< T >::is_forward().
|
inline |
Check if the index is valid or not.
Definition at line 81 of file Index.h.
Referenced by onert::ir::train::UseDefChain::insertTrainingDef().