ONE - On-device Neural Engine
|
#include <TestGraph.h>
Public Member Functions | |
TestGraph () | |
loco::Graph * | graph () |
template<class T > | |
T * | append () |
Creates node with NO arg and appends it to graph. | |
template<class T > | |
T * | append (loco::Node *arg1) |
Creates op T (arity=1) with arg1 as an input and appends it to graph. | |
template<class T > | |
T * | append (loco::Node *arg1, loco::Node *arg2) |
Creates op T (arity=2) with arg1, arg2 as inputs and appends it to graph. | |
template<class T > | |
T * | append (loco::Node *arg1, loco::Node *arg2, loco::Node *arg3) |
Creates op T (arity=3) with arg1, arg2, arg3 as inputs and appends it to graph. | |
void | complete () |
void | complete (loco::Node *last_node) |
Data Fields | |
std::unique_ptr< loco::Graph > | g |
loco::Pull * | pull |
loco::Push * | push |
Definition at line 33 of file TestGraph.h.
|
inline |
Definition at line 40 of file TestGraph.h.
References g, loco::link(), loco::make_graph(), pull, and push.
|
inline |
Creates node with NO arg and appends it to graph.
Definition at line 65 of file TestGraph.h.
References g.
|
inline |
Creates op T (arity=1) with arg1 as an input and appends it to graph.
Definition at line 74 of file TestGraph.h.
References g.
|
inline |
Creates op T (arity=2) with arg1, arg2 as inputs and appends it to graph.
Definition at line 84 of file TestGraph.h.
References g.
|
inline |
Creates op T (arity=3) with arg1, arg2, arg3 as inputs and appends it to graph.
Definition at line 94 of file TestGraph.h.
References g.
|
inline |
|
inline |
Definition at line 106 of file TestGraph.h.
References loco::Push::from(), and push.
|
inline |
Definition at line 62 of file TestGraph.h.
References g.
Referenced by TopologicalSortHelper.TopologicalSortHelper::add_edge(), and TopologicalSortHelper.TopologicalSortHelper::sort_util().
std::unique_ptr<loco::Graph> exo::test::TestGraph::g |
Definition at line 36 of file TestGraph.h.
Referenced by append(), append(), append(), append(), graph(), and TestGraph().
loco::Pull* exo::test::TestGraph::pull |
Definition at line 37 of file TestGraph.h.
Referenced by TestGraph().
loco::Push* exo::test::TestGraph::push |
Definition at line 38 of file TestGraph.h.
Referenced by complete(), complete(), and TestGraph().