|
ONE - On-device Neural Engine
|
Mapping between Feature/Tensor Axis. More...
#include <PermutingCodec.h>
Public Member Functions | |
| Permutation ()=default | |
| bool | mapped (const FeatureAxis &axis_f) const |
| Return whether a tensor axis is specified for a given feature axis. | |
| TensorAxis | axis (const FeatureAxis &axis_f) const |
| Get the tensor axis corresponding to a given feature axis. | |
| TensorAxis & | axis (const FeatureAxis &axis_f) |
| Set the tensor axis corresponding to a given feature axis. | |
| TensorAxis | operator[] (const FeatureAxis &axis_f) const |
| TensorAxis & | operator[] (const FeatureAxis &axis_f) |
Mapping between Feature/Tensor Axis.
Definition at line 44 of file PermutingCodec.h.
|
default |
| uint32_t & loco::Permutation< Domain::Feature >::axis | ( | const FeatureAxis & | axis_f | ) |
Set the tensor axis corresponding to a given feature axis.
Definition at line 100 of file PermutingCodec.cpp.
References loco::must_cast(), and loco::valid().
Referenced by onert.experimental.train.metrics.categorical_accuracy.CategoricalAccuracy::__init__(), and onert.experimental.train.metrics.categorical_accuracy.CategoricalAccuracy::update_state().
| uint32_t loco::Permutation< Domain::Feature >::axis | ( | const FeatureAxis & | axis_f | ) | const |
Get the tensor axis corresponding to a given feature axis.
This method works correclty only when feature axis is mapped before.
Definition at line 93 of file PermutingCodec.cpp.
References loco::must_cast(), and loco::valid().
Referenced by onert.experimental.train.metrics.categorical_accuracy.CategoricalAccuracy::__init__(), and onert.experimental.train.metrics.categorical_accuracy.CategoricalAccuracy::update_state().
| bool loco::Permutation< Domain::Feature >::mapped | ( | const FeatureAxis & | axis_f | ) | const |
Return whether a tensor axis is specified for a given feature axis.
This method does not validate the corresponding value.
Definition at line 87 of file PermutingCodec.cpp.
References loco::must_cast(), and loco::valid().
|
inline |
Definition at line 70 of file PermutingCodec.h.
References loco::must_cast().
|
inline |
Definition at line 69 of file PermutingCodec.h.
References loco::must_cast().