Use im2col from Caffe. More...

#include <neuron_layer.h>

Inheritance diagram for singa::CConvolutionLayer:

Public Member Functions
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...

Public Member Functions inherited from singa::ConvolutionLayer
void	Setup (const LayerProto &proto, int npartitions) override
	Setup layer properties. More...

const std::vector< Param * >	GetParams () const override
	Layers that have paramters must override this function. More...

ConnectionType	src_neuron_connection (int k) const override
	Return the connection type between one neuron of this layer and its source layer. More...

Public Member Functions inherited from singa::Layer
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)

Additional Inherited Members
Static Public Member Functions inherited from singa::Layer
static Layer *	Create (const LayerProto &proto)

Protected Attributes inherited from singa::ConvolutionLayer
int	kernel_

int	pad_

int	stride_

int	batchsize_

int	channels_

int	height_

int	width_

int	col_height_

int	col_width_

int	conv_height_

int	conv_width_

int	num_filters_

Param *	weight_

Param *	bias_

Blob< float >	col_data_

Blob< float >	col_grad_

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

Use im2col from Caffe.

Member Function Documentation

void singa::CConvolutionLayer::ComputeFeature	(	int	flag,
		Metric *	perf
	)

overridevirtual

Compute features of this layer based on connected layers.

Parameters

perf	pointer to Metric obj for collect and aggregate performance

Reimplemented from singa::ConvolutionLayer.

void singa::CConvolutionLayer::ComputeGradient	(	int	flag,
		Metric *	perf
	)

overridevirtual

Compute gradients for parameters and connected layers.

Parameters

flag	used to get the calling phase, e.g., forward of training (kForward \| kTrain)
flag	used to get the calling phase, e.g., forward of training

Reimplemented from singa::ConvolutionLayer.

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

/home/wangwei/program/asf/release-0.1/apache-singa-incubating-0.1.0-RC1/include/neuralnet/neuron_layer.h

Public Member Functions

Additional Inherited Members

Detailed Description

Member Function Documentation