|
| P2PSync (shared_ptr< Solver< Dtype > > root_solver, P2PSync< Dtype > *parent, const SolverParameter ¶m) |
|
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 |
|
| GPUParams (shared_ptr< Solver< Dtype > > root_solver, int device) |
|
void | configure (Solver< Dtype > *solver) const |
|
| Params (shared_ptr< Solver< Dtype > > root_solver) |
|
size_t | size () const |
|
Dtype * | data () const |
|
Dtype * | diff () const |
|
void | StartInternalThread () |
|
void | StopInternalThread () |
|
bool | is_started () const |
|
|
void | on_start () |
|
void | on_gradients_ready () |
|
void | InternalThreadEntry () |
|
| DISABLE_COPY_AND_ASSIGN (Params) |
|
bool | must_stop () |
|
|
P2PSync< Dtype > * | parent_ |
|
vector< P2PSync< Dtype > * > | children_ |
|
BlockingQueue< P2PSync< Dtype > * > | queue_ |
|
const int | initial_iter_ |
|
Dtype * | parent_grads_ |
|
shared_ptr< Solver< Dtype > > | solver_ |
|
const size_t | size_ |
|
Dtype * | data_ |
|
Dtype * | diff_ |
|
The documentation for this class was generated from the following files: