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

Base layer for connecting layers when neural net is partitioned. More...

#include <layer.h>

Inheritance diagram for singa::ConnectionLayer:
singa::Layer singa::BridgeLayer singa::ConcateLayer singa::SliceLayer singa::SplitLayer singa::BridgeDstLayer singa::BridgeSrcLayer

Additional Inherited Members

- 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)
 
- Static Public Member Functions inherited from singa::Layer
static LayerCreate (const LayerProto &proto)
 
- Protected Attributes inherited from singa::Layer
LayerProto layer_proto_
 
Blob< float > data_
 
Blob< float > grad_
 
std::vector< Layer * > srclayers_
 
std::vector< Layer * > dstlayers_
 

Detailed Description

Base layer for connecting layers when neural net is partitioned.


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