ONE - On-device Neural Engine
Loading...
Searching...
No Matches
loco::TensorIndex Class Reference

#include <TensorIndex.h>

Public Member Functions

 TensorIndex ()
 
uint32_t rank (void) const
 
TensorIndexresize (uint32_t size)
 
uint32_tat (uint32_t axis)
 
uint32_t at (uint32_t axis) const
 

Detailed Description

Definition at line 27 of file TensorIndex.h.

Constructor & Destructor Documentation

◆ TensorIndex()

loco::TensorIndex::TensorIndex ( )
default

Member Function Documentation

◆ at() [1/2]

◆ at() [2/2]

uint32_t loco::TensorIndex::at ( uint32_t  axis) const

Definition at line 34 of file TensorIndex.cpp.

34{ return _indices.at(axis); }

◆ rank()

uint32_t loco::TensorIndex::rank ( void  ) const

Definition at line 24 of file TensorIndex.cpp.

24{ return _indices.size(); }

Referenced by loco::PermutingDecoder< Domain::DepthwiseFilter >::value().

◆ resize()


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