ONE - On-device Neural Engine
Loading...
Searching...
No Matches
GraphTestcase< GraphCode::ConstGen > Class Referencefinal

#include <GraphTestcase.h>

Collaboration diagram for GraphTestcase< GraphCode::ConstGen >:

Public Member Functions

 GraphTestcase ()
 
loco::Graphgraph ()
 

Data Fields

loco::ConstGenconst_node = nullptr
 
loco::Pushpush_node = nullptr
 

Detailed Description

Definition at line 120 of file GraphTestcase.h.

Constructor & Destructor Documentation

◆ GraphTestcase()

Definition at line 123 of file GraphTestcase.h.

124 {
125 _graph = loco::make_graph();
126
127 auto graph_builder = make_graph_builder(_graph.get());
128
129 const_node = graph_builder->push<ConstGenLayer>()->node();
130
131 push_node = graph_builder->push<OutputLayer>()->name("output")->node();
132 }
std::unique_ptr< Graph > make_graph(void)
Definition Graph.cpp:131

References loco::make_graph().

Member Function Documentation

◆ graph()

Field Documentation

◆ const_node

Definition at line 137 of file GraphTestcase.h.

◆ push_node

loco::Push* GraphTestcase< GraphCode::ConstGen >::push_node = nullptr

Definition at line 138 of file GraphTestcase.h.


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