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

Public Member Functions

def __init__ (self, name, max_concurrent_context, param_init_net, rendezvous)
 
def get_control_and_context (self, control_output_blob)
 

Public Attributes

 name
 
 param_init_net
 
 max_concurrent_context
 
 counter
 
 common_worlds
 
 control_inputs
 
 rendezvous
 

Detailed Description

Creates common worlds (up to max_concurrent_context) and manage the
sequential execution of collectives that shares the same context with
cyclic control inputs.

Definition at line 1303 of file data_parallel_model.py.


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