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

Create a feature map from a tensor. More...

#include <Nodes.h>

Collaboration diagram for loco::FeatureEncode:
Collaboration graph
[legend]

Public Member Functions

Node * input (void) const
 
void input (Node *node)
 
FeatureEncoder * encoder (void) const
 
void encoder (std::unique_ptr< FeatureEncoder > &&enc)
 
- Public Member Functions inherited from loco::CanonicalNodeDef< CanonicalOpcode::FeatureEncode, FixedArity< 1 >::Mixin >
virtual ~CanonicalNodeDef ()=default
 
uint32_t opnum (void) const final
 
CanonicalOpcode opcode (void) const final
 
- Public Member Functions inherited from loco::CanonicalNode
virtual ~CanonicalNode ()=default
 
const Dialect * dialect (void) const final
 Return "Dialect" identifier that this node belongs to.
 
template<typename T >
T accept (CanonicalNodeVisitorBase< T > *) const
 
template<typename T >
T accept (CanonicalNodeMutableVisitorBase< T > *)
 
- Public Member Functions inherited from loco::Node
 Node ()=default
 
 Node (const Node &)=delete
 
 Node (Node &&)=delete
 
virtual ~Node ()
 
Graph * graph (void)
 
const Graph * graph (void) const
 
virtual uint32_t arity (void) const =0
 Return the number of arguments.
 
virtual Node * arg (uint32_t N) const =0
 Access N-th argument node.
 
virtual void drop (void)=0
 Drop all the reference of arguments.
 
- Public Member Functions inherited from loco::AnnotatedItem< NodeAnnotation >
 AnnotatedItem ()=default
 
virtual ~AnnotatedItem ()=default
 
const T * annot (void) const
 Retrieve a stored annotation of type T.
 
void annot (std::unique_ptr< T > &&p)
 Attach or remove a new annotation of type T.
 

Detailed Description

Create a feature map from a tensor.

Definition at line 378 of file Nodes.h.

Member Function Documentation

◆ encoder() [1/2]

void loco::FeatureEncode::encoder ( std::unique_ptr< FeatureEncoder > &&  enc)
inline

Definition at line 387 of file Nodes.h.

387{ _enc = std::move(enc); }
loco::must_cast
T must_cast(FeatureEncoder *node)
A helper dynamic_cast that throws when failed.
Definition FeatureCodec.h:80

References loco::must_cast().

◆ encoder() [2/2]

FeatureEncoder * loco::FeatureEncode::encoder ( void  ) const
inline

Definition at line 386 of file Nodes.h.

386{ return _enc.get(); }

Referenced by FeatureEncodeLayer::Return::perm(), and logo::SimplifyDomainConversionPass::run().

◆ input() [1/2]

void loco::FeatureEncode::input ( Node *  node)
inline

Definition at line 383 of file Nodes.h.

383{ at(0)->node(node); }

◆ input() [2/2]

Node * loco::FeatureEncode::input ( void  ) const
inline

Definition at line 382 of file Nodes.h.

382{ return at(0)->node(); }

Referenced by GraphTestcase< GraphCode::AvgPool2D >::GraphTestcase(), GraphTestcase< GraphCode::MaxPool2D >::GraphTestcase(), exo::make_feature_encode(), FeatureEncodeLayer::operator()(), logo::SimplifyDomainConversionPass::run(), mir2loco::Transformer::visit(), mir2loco::Transformer::visit(), mir2loco::Transformer::visit(), mir2loco::Transformer::visit(), and mir2loco::Transformer::visit().

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