Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
nom::repr::NNModule Struct Reference

Public Member Functions

 NNModule (const NNModule &)=delete
 
 NNModule (NNModule &&)=default
 
void replaceSubgraph (const NNGraph::SubgraphType &subgraph, const NNGraph::NodeRef &node, const std::vector< NNGraph::NodeRef > &node_inputs, const std::vector< NNGraph::NodeRef > &node_outputs)
 
void deleteSubgraph (const NNGraph::SubgraphType &subgraph)
 
NNGraph::NodeRef createUniqueDataNode (const std::string &s="_unique")
 
template<typename T , typename... Args>
NNGraph::NodeRef replaceSubgraphWithOperator (const NNGraph::SubgraphType &, const std::vector< NNGraph::NodeRef > &, const std::vector< NNGraph::NodeRef > &, Args...)
 

Data Fields

NNGraph dataFlow
 
NNCFGraph controlFlow
 
std::unordered_set< NNGraph::NodeRefinputs
 
std::unordered_set< NNGraph::NodeRefoutputs
 

Detailed Description

Definition at line 247 of file NeuralNet.h.


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