ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Public Member Functions
loco::TensorAxisSet Class Referencefinal

#include <TensorAxisSet.h>

Public Member Functions

 TensorAxisSet ()=default
 
bool defined (const TensorAxis &axis) const
 
void insert (const TensorAxis &axis)
 

Detailed Description

Definition at line 27 of file TensorAxisSet.h.

Constructor & Destructor Documentation

◆ TensorAxisSet()

loco::TensorAxisSet::TensorAxisSet ( )
default

Member Function Documentation

◆ defined()

bool loco::TensorAxisSet::defined ( const TensorAxis &  axis) const
inline

Definition at line 33 of file TensorAxisSet.h.

33{ return _axes.find(axis) != _axes.end(); }

◆ insert()

void loco::TensorAxisSet::insert ( const TensorAxis &  axis)
inline

Definition at line 34 of file TensorAxisSet.h.

34{ _axes.insert(axis); }
The documentation for this class was generated from the following file: