|
void | Setup (const LayerProto &proto, int npartitions) override |
| Setup layer properties. More...
|
|
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...
|
|
const std::vector< Param * > | GetParams () const override |
| 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) |
|