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 (luci::CircleNode *arg1) |
Creates op T (arity=1) with arg1 as an input and appends it to graph. | |
template<class T > | |
T * | append (luci::CircleNode *arg1, luci::CircleNode *arg2) |
Creates op T (arity=2) with arg1, arg2 as inputs and appends it to graph. | |
template<class T > | |
T * | append (luci::CircleNode *arg1, luci::CircleNode *arg2, luci::CircleNode *arg3) |
Creates op T (arity=3) with arg1, arg2, arg3 as inputs and appends it to graph. | |
void | complete () |
void | complete (luci::CircleNode *last_node) |
Data Fields | |
std::unique_ptr< loco::Graph > | g |
luci::CircleInput * | input_node = nullptr |
luci::CircleOutput * | output_node = nullptr |
Definition at line 34 of file TestGraph.h.
|
inline |
Definition at line 41 of file TestGraph.h.
References g, input_node, luci::link(), loco::make_graph(), and output_node.
|
inline |
Creates node with NO arg and appends it to graph.
Definition at line 66 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 75 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 85 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 96 of file TestGraph.h.
References g.
|
inline |
|
inline |
|
inline |
Definition at line 63 of file TestGraph.h.
References g.
Referenced by TopologicalSortHelper.TopologicalSortHelper::add_edge(), and TopologicalSortHelper.TopologicalSortHelper::sort_util().
std::unique_ptr<loco::Graph> luci::test::TestGraph::g |
Definition at line 37 of file TestGraph.h.
Referenced by append(), append(), append(), append(), graph(), and TestGraph().
luci::CircleInput* luci::test::TestGraph::input_node = nullptr |
Definition at line 38 of file TestGraph.h.
Referenced by TestGraph().
luci::CircleOutput* luci::test::TestGraph::output_node = nullptr |
Definition at line 39 of file TestGraph.h.
Referenced by complete(), complete(), and TestGraph().