Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
torch::jit::TopoMoveTestFixture Struct Reference

Public Member Functions

void createGraph ()
 
void createNode (const std::string &name, const std::vector< std::string > &inputNames, const std::vector< std::string > &blockInputNames={})
 
bool moveBeforeTopologicallyValid (const std::string &toInsert, const std::string &insertPoint)
 
bool moveAfterTopologicallyValid (const std::string &toInsert, const std::string &insertPoint)
 
bool moveWithChecks (const std::string &toInsert, const std::string &insertPoint, std::function< bool(Node *, Node *)> func)
 
void checkPostCondition (const std::string &toInsert, const std::string &insertPoint, bool after)
 

Data Fields

std::shared_ptr< Graphgraph
 
std::unique_ptr< AliasDbaliasDb
 
std::unordered_map< std::string, Node * > nodes
 

Detailed Description

Definition at line 13 of file test_alias_analysis.h.


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