ONE - On-device Neural Engine
Loading...
Searching...
No Matches
FeatureDecodeLayer::Return Class Reference

#include <GraphBuilder.h>

Public Member Functions

 Return (loco::FeatureDecode *node)
 
Returnperm (const loco::Permutation< loco::Domain::Feature > &perm)
 
loco::FeatureDecodenode (void)
 

Detailed Description

Definition at line 309 of file GraphBuilder.h.

Constructor & Destructor Documentation

◆ Return()

FeatureDecodeLayer::Return::Return ( loco::FeatureDecode node)
inline

Definition at line 312 of file GraphBuilder.h.

312 : _node{node}
313 {
314 // DO NOTHING
315 }
loco::FeatureDecode * node(void)

Member Function Documentation

◆ node()

loco::FeatureDecode * FeatureDecodeLayer::Return::node ( void  )
inline

Definition at line 326 of file GraphBuilder.h.

326{ return _node; }

◆ perm()

Return * FeatureDecodeLayer::Return::perm ( const loco::Permutation< loco::Domain::Feature > &  perm)
inline

Definition at line 318 of file GraphBuilder.h.

319 {
320 using namespace loco;
321 _node->decoder(std::make_unique<PermutingDecoder<Domain::Feature>>(perm));
322 return this;
323 }
Return * perm(const loco::Permutation< loco::Domain::Feature > &perm)
FeatureDecoder * decoder(void) const
Definition Nodes.h:405

References loco::FeatureDecode::decoder(), and perm().

Referenced by perm().


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