ONE - On-device Neural Engine
Loading...
Searching...
No Matches
luci::tinf::Algorithm Class Referencefinal

#include <CircleTypeInference.h>

Collaboration diagram for luci::tinf::Algorithm:

Public Member Functions

loco::DataType visit (const luci::CircleNode *node) final
 Default fallback.
 
loco::DataType visit (const luci::CircleIfOut *node) final
 
- Public Member Functions inherited from luci::CircleNodeVisitor< loco::DataType >
virtual ~CircleNodeVisitor ()=default
 
- Public Member Functions inherited from luci::CircleNodeVisitorBase< T >
virtual ~CircleNodeVisitorBase ()=default
 

Detailed Description

Definition at line 37 of file CircleTypeInference.h.

Member Function Documentation

◆ visit() [1/2]

loco::DataType luci::tinf::Algorithm::visit ( const luci::CircleIfOut node)
final

Definition at line 88 of file CircleIfOut.cpp.

89{
90 auto graphs = get_out_graphs(node);
91 assert(graphs.then_graph_output->dtype() == graphs.else_graph_output->dtype());
92 return graphs.then_graph_output->dtype();
93}

◆ visit() [2/2]

loco::DataType luci::tinf::Algorithm::visit ( const luci::CircleNode )
inlinefinalvirtual

Default fallback.

Reimplemented from luci::CircleNodeVisitor< loco::DataType >.

Definition at line 41 of file CircleTypeInference.h.

42 {
45 return dtype;
46 }
DataType
"scalar" value type
Definition DataType.h:27
Type Inference Rule for CircleDialect.
bool infer(const loco::Node *, loco::DataType &) const final

References luci::CircleTypeInferenceRule::infer().


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