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

Public Member Functions

def __init__ (self)
 
def add_node (self, node)
 
def nodes (self)
 
def node_kwargs (self)
 
def __repr__ (self)
 

Detailed Description

Context that keeps track of all the node names used.
Users shouldn't have to use them directly, since a Cluster is automatically
generated at the first usage of 'Node'.

Definition at line 27 of file task.py.

Member Function Documentation

def caffe2.python.task.Cluster.nodes (   self)
Returns the list of unique node names used within this context.

Definition at line 46 of file task.py.


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