Apache SINGA
A distributed deep learning platform .
 All Classes Namespaces Files Functions Variables Typedefs Macros
Public Member Functions | Protected Attributes | List of all members
singa::BridgeLayer Class Reference
Inheritance diagram for singa::BridgeLayer:
singa::ConnectionLayer singa::Layer singa::BridgeDstLayer singa::BridgeSrcLayer

Public Member Functions

void set_ready (bool a)
 
bool ready () const
 
virtual bool is_bridgesrclayer () const
 
virtual bool is_bridgedstlayer () const
 
- Public Member Functions inherited from singa::Layer
virtual void Setup (const LayerProto &proto, int npartitions=1)
 Setup layer properties. More...
 
virtual void ComputeFeature (int flag, Metric *perf)=0
 Compute features of this layer based on connected layers. More...
 
virtual void ComputeGradient (int flag, Metric *perf)=0
 Compute gradients for parameters and connected layers. More...
 
virtual const std::vector
< Param * > 
GetParams () const
 Layers that have paramters must override this function. More...
 
virtual ConnectionType src_neuron_connection (int k) const
 Return the connection type between one neuron of this layer and its source layer. More...
 
virtual ConnectionType dst_layer_connection () const
 Return the connection type of this layer and all dst layers. More...
 
virtual const std::string DebugString (int step, int flag)
 For print debug info about each layer, e.g., norm of feature vector, norm of parameters. More...
 
int partition_dim () const
 
int partition_id () const
 
int type () const
 
const std::string & name () const
 Return name of this layer.
 
virtual const Blob< float > & data (const Layer *from) const
 
virtual Blob< float > * mutable_data (const Layer *from)
 
virtual const Blob< float > & grad (const Layer *from) const
 
virtual Blob< float > * mutable_grad (const Layer *from)
 
const std::vector< Layer * > srclayers () const
 return LayerS that connected to this layer
 
const std::vector< Layer * > dstlayers () const
 return LayerS that this layer connected to
 
int srclayers_size () const
 
int dstlayers_size () const
 
void clear_dstlayers ()
 
void clear_srclayers ()
 
void add_srclayer (Layer *src)
 
void add_dstlayer (Layer *dst)
 

Protected Attributes

bool ready_
 < true if received grad from BridgeDstLayer
 
- Protected Attributes inherited from singa::Layer
LayerProto layer_proto_
 
Blob< float > data_
 
Blob< float > grad_
 
std::vector< Layer * > srclayers_
 
std::vector< Layer * > dstlayers_
 

Additional Inherited Members

- Static Public Member Functions inherited from singa::Layer
static LayerCreate (const LayerProto &proto)
 

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