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

#include <GraphBuilder.h>

Public Member Functions

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

Detailed Description

Definition at line 272 of file GraphBuilder.h.

Constructor & Destructor Documentation

◆ Return()

FeatureEncodeLayer::Return::Return ( loco::FeatureEncode node)
inline

Definition at line 275 of file GraphBuilder.h.

275 : _node{node}
276 {
277 // DO NOTHING
278 }
loco::FeatureEncode * node(void)

Member Function Documentation

◆ node()

loco::FeatureEncode * FeatureEncodeLayer::Return::node ( void  )
inline

Definition at line 289 of file GraphBuilder.h.

289{ return _node; }

◆ perm()

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

Definition at line 281 of file GraphBuilder.h.

282 {
283 using namespace loco;
284 _node->encoder(std::make_unique<PermutingEncoder<Domain::Feature>>(perm));
285 return this;
286 }
Return * perm(const loco::Permutation< loco::Domain::Feature > &perm)
FeatureEncoder * encoder(void) const
Definition Nodes.h:386

References loco::FeatureEncode::encoder(), and perm().

Referenced by perm().


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