ONE - On-device Neural Engine
Loading...
Searching...
No Matches
loco::TensorBroadcast::Mapping Class Referencefinal

#include <Nodes.h>

Public Member Functions

 Mapping ()=default
 
bool defined (const TensorAxis &axis) const
 
const Dimensiondim (const TensorAxis &axis) const
 
Dimensiondim (const TensorAxis &axis)
 

Detailed Description

Definition at line 989 of file Nodes.h.

Constructor & Destructor Documentation

◆ Mapping()

loco::TensorBroadcast::Mapping::Mapping ( )
default

Member Function Documentation

◆ defined()

bool loco::TensorBroadcast::Mapping::defined ( const TensorAxis axis) const

Definition at line 231 of file Nodes.cpp.

232{
233 return _content.find(axis) != _content.end();
234}

◆ dim() [1/2]

Dimension & loco::TensorBroadcast::Mapping::dim ( const TensorAxis axis)

Definition at line 241 of file Nodes.cpp.

241{ return _content[axis]; }

◆ dim() [2/2]

const Dimension & loco::TensorBroadcast::Mapping::dim ( const TensorAxis axis) const

Definition at line 236 of file Nodes.cpp.

237{
238 return _content.at(axis);
239}

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