Caffe
Public Member Functions | Protected Member Functions | Protected Attributes | List of all members
caffe::P2PSync< Dtype > Class Template Reference
Inheritance diagram for caffe::P2PSync< Dtype >:
caffe::GPUParams< Dtype > caffe::Solver< Dtype >::Callback caffe::InternalThread caffe::Params< Dtype >

Public Member Functions

 P2PSync (shared_ptr< Solver< Dtype > > root_solver, P2PSync< Dtype > *parent, const SolverParameter &param)
 
const shared_ptr< Solver< Dtype > > & solver () const
 
void Run (const vector< int > &gpus)
 
void Prepare (const vector< int > &gpus, vector< shared_ptr< P2PSync< Dtype > > > *syncs)
 
const int initial_iter () const
 
- Public Member Functions inherited from caffe::GPUParams< Dtype >
 GPUParams (shared_ptr< Solver< Dtype > > root_solver, int device)
 
void configure (Solver< Dtype > *solver) const
 
- Public Member Functions inherited from caffe::Params< Dtype >
 Params (shared_ptr< Solver< Dtype > > root_solver)
 
size_t size () const
 
Dtype * data () const
 
Dtype * diff () const
 
- Public Member Functions inherited from caffe::InternalThread
void StartInternalThread ()
 
void StopInternalThread ()
 
bool is_started () const
 

Protected Member Functions

void on_start ()
 
void on_gradients_ready ()
 
void InternalThreadEntry ()
 
- Protected Member Functions inherited from caffe::Params< Dtype >
 DISABLE_COPY_AND_ASSIGN (Params)
 
- Protected Member Functions inherited from caffe::InternalThread
bool must_stop ()
 

Protected Attributes

P2PSync< Dtype > * parent_
 
vector< P2PSync< Dtype > * > children_
 
BlockingQueue< P2PSync< Dtype > * > queue_
 
const int initial_iter_
 
Dtype * parent_grads_
 
shared_ptr< Solver< Dtype > > solver_
 
- Protected Attributes inherited from caffe::Params< Dtype >
const size_t size_
 
Dtype * data_
 
Dtype * diff_
 

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