Apache SINGA
A distributed deep learning platform .
 All Classes Namespaces Files Functions Variables Typedefs Enumerator Macros
Public Member Functions | List of all members
singa::RGBImageLayer Class Reference
Inheritance diagram for singa::RGBImageLayer:
singa::ParserLayer singa::Layer

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_

Member Function Documentation

virtual void singa::RGBImageLayer::ParseRecords ( bool  training,
const vector< Record > &  records,
Blob< float > *  blob 

Parse records from DataLayer into blob.

This function is called by ComputeFeature(bool, const vector<SLayer>& srclayers) or Prefetch(bool).

Implements singa::ParserLayer.

virtual void singa::RGBImageLayer::Setup ( const LayerProto proto,
const vector< SLayer > &  srclayers 

Setup layer properties.

Setup the shapes for data and parameters, also setup some properties based on the layer configuration and connected src layers.

srclayerslayers connecting to this layer

Implements singa::ParserLayer.

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