ONE - On-device Neural Engine
|
#include <TestHelper.h>
Public Member Functions | |
TFNodeBuildTester () | |
void | inputs (const std::vector< std::string > &names) |
void | inputs (const std::vector< std::string > &names, const loco::DataType dtype) |
void | output (const char *name) |
moco::TFNode * | output (void) |
void | run (tensorflow::NodeDef &node_def, moco::GraphBuilder &graph_builder) |
Definition at line 59 of file TestHelper.h.
moco::test::TFNodeBuildTester::TFNodeBuildTester | ( | ) |
void moco::test::TFNodeBuildTester::inputs | ( | const std::vector< std::string > & | names | ) |
void moco::test::TFNodeBuildTester::inputs | ( | const std::vector< std::string > & | names, |
const loco::DataType | dtype | ||
) |
void moco::test::TFNodeBuildTester::output | ( | const char * | name | ) |
moco::TFNode * moco::test::TFNodeBuildTester::output | ( | void | ) |
void moco::test::TFNodeBuildTester::run | ( | tensorflow::NodeDef & | node_def, |
moco::GraphBuilder & | graph_builder | ||
) |