ONE - On-device Neural Engine
Loading...
Searching...
No Matches
loco::TensorTranspose::Perm Class Referencefinal

#include <Nodes.h>

Public Member Functions

 Perm ()=default
 
uint32_t size () const
 
void size (uint32_t size)
 
const TensorAxisaxis (TensorAxis n) const
 
TensorAxisaxis (TensorAxis n)
 

Detailed Description

Definition at line 1098 of file Nodes.h.

Constructor & Destructor Documentation

◆ Perm()

loco::TensorTranspose::Perm::Perm ( )
default

Member Function Documentation

◆ axis() [1/2]

TensorAxis & loco::TensorTranspose::Perm::axis ( TensorAxis  n)
inline

Definition at line 1108 of file Nodes.h.

1108{ return _vals[n]; }
T must_cast(FeatureEncoder *node)
A helper dynamic_cast that throws when failed.

References loco::must_cast().

Referenced by onert.experimental.train.metrics.categorical_accuracy.CategoricalAccuracy::__init__(), and onert.experimental.train.metrics.categorical_accuracy.CategoricalAccuracy::update_state().

◆ axis() [2/2]

◆ size() [1/2]

uint32_t loco::TensorTranspose::Perm::size ( ) const
inline

Definition at line 1104 of file Nodes.h.

1104{ return _vals.size(); }

Referenced by exo::TensorTransposeConverter::convert(), logo::SimplifyDomainConversionPass::run(), and size().

◆ size() [2/2]

void loco::TensorTranspose::Perm::size ( uint32_t  size)
inline

Definition at line 1105 of file Nodes.h.

1105{ _vals.resize(size); }
uint32_t size() const
Definition Nodes.h:1104

References size().


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