Apache SINGA
A distributed deep learning platform .
 All Classes Namespaces Files Functions Variables Typedefs Macros
Public Member Functions | Public Attributes | List of all members
singa::ParamEntry Class Reference

ParamEntry is used for aggregating gradients of Params shared by workers from the same group. More...

#include <param.h>

Public Member Functions

 ParamEntry (int total, Param *p)
 
void AddParam (bool local, Param *p)
 Associate the counter to a Param object. More...
 

Public Attributes

int next_version = -1
 
int num_update = 0
 
int num_local = 0
 

local workers using the shared parameter


 
int num_total = 0
 

total workers using the shared parameter

More...
 
std::vector< Param * > shares
 

Detailed Description

ParamEntry is used for aggregating gradients of Params shared by workers from the same group.

For each worker group, every unique Param object has a ParamEntry object. Param objects sharing the same values are associated with the same ParamEntry.

Member Function Documentation

void singa::ParamEntry::AddParam ( bool  local,
Param p 
)

Associate the counter to a Param object.

Parameters
p
local1 if it is used by workers in this procs, 0 otherwise

Member Data Documentation

int singa::ParamEntry::num_total = 0

total workers using the shared parameter

Shares are deleted by neuralnet's destructor


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