ONE - On-device Neural Engine
|
Data Structures | |
class | TFNodeBuildTester |
Functions | |
template<typename T > | |
T * | find_first_node_bytype (loco::Graph *g) |
template<typename T > | |
std::vector< T * > | find_nodes_bytype (loco::Graph *g) |
void | setup_output_node (loco::Graph *graph, loco::Node *last_node) |
Append setup output of graph by adding loco::Push node. | |
T * moco::tf::test::find_first_node_bytype | ( | loco::Graph * | g | ) |
Definition at line 33 of file TestHelper.h.
References loco::ObjectPool< T >::at(), and loco::ObjectPool< T >::size().
std::vector< T * > moco::tf::test::find_nodes_bytype | ( | loco::Graph * | g | ) |
Definition at line 49 of file TestHelper.h.
References loco::ObjectPool< T >::at(), and loco::ObjectPool< T >::size().
void moco::tf::test::setup_output_node | ( | loco::Graph * | graph, |
loco::Node * | last_node | ||
) |
Append setup output of graph by adding loco::Push node.