 |
Apache SINGA
A distributed deep learning platform .
|
|
Apache SINGA
A distributed deep learning platform .
|
Derived from ParserLayer to parse label from SingaleLabelImageRecord. More...
#include <input_layer.h>
Public Member Functions | |
| void | Setup (const LayerProto &proto, int npartitions) override |
| Setup layer properties. More... | |
| void | ParseRecords (int flag, const std::vector< Record > &records, Blob< float > *blob) override |
| Parse records from DataLayer into blob. | |
 Public Member Functions inherited from singa::ParserLayer | |
| void | ComputeFeature (int flag, Metric *perf) override |
| Compute features of this layer based on connected layers. More... | |
| void | ComputeGradient (int flag, Metric *perf) override |
| Compute gradients for parameters and connected layers. More... | |
| ConnectionType | dst_layer_connection () const override |
| Return the connection type of this layer and all dst layers. More... | |
| Blob< float > * | mutable_grad (const Layer *layer) override |
| const Blob< float > & | grad (const Layer *from) const override |
| Public Member Functions inherited from singa::Layer | |
| 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 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) |
| 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) |
Additional Inherited Members | |
| Static Public Member Functions inherited from singa::Layer | |
| static Layer * | Create (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_ |
Derived from ParserLayer to parse label from SingaleLabelImageRecord.
|
overridevirtual |
Setup layer properties.
Setup the shapes for data and parameters, also setup some properties based on the layer configuration and connected layers.
| proto | layer configuration. |
| npartitions | num of total partitions of the original layer. This layer should be setup as one partition. |
Reimplemented from singa::Layer.
1.8.6
