Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | List of all members
caffe2.python.nomnigraph.NNModule Class Reference
Inheritance diagram for caffe2.python.nomnigraph.NNModule:

Public Member Functions

def __init__ (self, net=None, device_map=None)
 
def dataFlow (self)
 
def controlFlow (self)
 
def nodes (self)
 
def operators (self)
 
def tensors (self)
 
def createNode (self, val)
 
def deleteNode (self, node)
 
def createEdge (self, a, b)
 
def deleteEdge (self, a, b=None)
 
def replaceNode (self, old_node, new_node)
 
def replaceProducer (self, tensor, new_producer)
 
def replaceAllUsesWith (self, old_tensor, new_tensor)
 
def replaceAsConsumer (self, old_consumer, new_consumer)
 
def replaceSubgraph (self, subgraph, new_node, inputs, outputs)
 
def deleteSubgraph (self, subgraph)
 
def createUniqueDataNode (self, prefix="_unique")
 
def convertToCaffe2Proto (self, old_proto=None)
 
def match (self, pattern)
 

Detailed Description

Definition at line 14 of file nomnigraph.py.


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