 |
Apache SINGA
A distributed deep learning platform .
|
|
Apache SINGA
A distributed deep learning platform .
|
Public Member Functions | |
| virtual void | Setup (const LayerProto &proto, const vector< SLayer > &srclayers) |
| Setup layer properties. More... | |
| virtual void | ParseRecords (bool training, const vector< Record > &records, Blob< float > *blob) |
| Parse records from DataLayer into blob. More... | |
 Public Member Functions inherited from singa::ParserLayer | |
| virtual bool | is_parserlayer () const |
| virtual void | ComputeFeature (bool training, const vector< SLayer > &srclayers) |
| Compute features of this layer based on connected layers. More... | |
| virtual void | ComputeGradient (const vector< SLayer > &srclayers) |
| Dummy function. More... | |
| virtual void | Setup () |
| Setup layer properties. More... | |
| virtual void | SetupAfterPartition () |
| Setup the layer properties except shape. | |
| virtual void | SetupAfterPartition (const LayerProto &proto, const vector< int > &shape, const vector< SLayer > &srclayers) |
| Setup the layer properties except shape. More... | |
| virtual PartitionType | partition_type () const |
| virtual Blob< float > * | mutable_grad (const Layer *layer) |
| virtual const Blob< float > & | grad (const Layer *from) const |
| Public Member Functions inherited from singa::Layer | |
| virtual void | Init (const LayerProto &proto) |
| Layer initialization. More... | |
| void | Init (const Layer &other, const vector< int > &shape) |
| Copy layer configuration from the other Layer, and use the shape argument to as its data shape. | |
| virtual void | ToProto (LayerProto *layer_proto, bool copyData) |
| TODO(wangsheng) Marshal layer properties and data into google protobuf object (i.e., snapshot). More... | |
| virtual vector< shared_ptr < Param > > | GetParams () |
| Layers that have paramters must overload this function. More... | |
| virtual void | ComputeFeature (bool training) |
| virtual void | ComputeGradient () |
| Compute gradients for parameters and connecting layers. | |
| virtual int | partition_dimension () const |
| Decide on which dimension to do the partitioning. More... | |
| virtual ConnectionType | connection_type (int k) const |
| Return connection type between two layers. More... | |
| virtual void | set_partitionid (int id) |
| partition id is the ID of the layer in the original layer. | |
| virtual int | partitionid () const |
| virtual void | set_name (string name) |
| virtual const string | type () const |
| const std::string & | name () const |
| Return name of this layer. | |
| const std::string & | datablob () const |
| const vector< int > & | shape (const Layer *layer) const |
| virtual const Blob< float > & | data (const Layer *from) const |
| virtual Blob< float > * | mutable_data (const Layer *from) |
| virtual const vector< SLayer > | srclayers () const |
| return LayerS that connected to this layer | |
| virtual const vector< SLayer > | dstlayers () const |
| return LayerS that this layer connected to | |
| virtual const int | srclayers_size () const |
| virtual const int | dstlayers_size () const |
| virtual void | ClearDstLayers () |
| virtual void | ClearSrcLayers () |
| virtual void | AddSrcLayer (SLayer src) |
| virtual void | AddDstLayer (SLayer dst) |
| virtual bool | is_datalayer () const |
| virtual bool | is_losslayer () const |
| virtual bool | is_bridgesrclayer () const |
| virtual bool | is_bridgedstlayer () const |
Additional Inherited Members | |
| Protected Attributes inherited from singa::Layer | |
| string | name_ |
| Blob< float > | data_ |
| Blob< float > | grad_ |
| LayerProto | layer_proto_ |
| vector< SLayer > | srclayers_ |
| vector< SLayer > | dstlayers_ |
|
virtual |
Parse records from DataLayer into blob.
This function is called by ComputeFeature(bool, const vector<SLayer>& srclayers) or Prefetch(bool).
Implements singa::ParserLayer.
|
virtual |
Setup layer properties.
Setup the shapes for data and parameters, also setup some properties based on the layer configuration and connected src layers.
| srclayers | layers connecting to this layer |
Implements singa::ParserLayer.
1.8.6
