For sending data to layer on other threads which may resident on other nodes due to layer/data partition. More...

#include <connection_layer.h>

Inheritance diagram for singa::BridgeSrcLayer:

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...

const Blob< float > &	data (const Layer *from) const override

Blob< float > *	mutable_data (const Layer *from) override

const Blob< float > &	grad (const Layer *from) const override

Blob< float > *	mutable_grad (const Layer *from) override

bool	is_bridgesrclayer () const override

Public Member Functions inherited from singa::BridgeLayer
void	set_ready (bool a)

bool	ready () const

virtual bool	is_bridgedstlayer () const

Public Member Functions inherited from singa::Layer
virtual void	Setup (const LayerProto &proto, int npartitions=1)
	Setup layer properties. More...

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 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.

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::BridgeLayer
bool	ready_
	< true if received grad from BridgeDstLayer

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

For sending data to layer on other threads which may resident on other nodes due to layer/data partition.

Member Function Documentation

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

inlineoverridevirtual

Compute features of this layer based on connected layers.

Parameters

perf	pointer to Metric obj for collect and aggregate performance

Implements singa::Layer.

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

inlineoverridevirtual

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

Implements singa::Layer.

const Blob<float>& singa::BridgeSrcLayer::data ( const Layer * from ) const

inlineoverridevirtual

Returns: name of src data blob, used by prefetch layer to locate the data blob in parser layers; The default value is "unknown"; If the src layer is the prefetch layer and there are more than one parser layers, this value be set. const std::string &datablob() const { return layer_proto_.datablob(); }; a const ref for Blob storing neuron values of this layer for BP

Reimplemented from singa::Layer.

Blob<float>* singa::BridgeSrcLayer::mutable_grad ( const Layer * from )

inlineoverridevirtual

Returns: a pointer to storing neuron grads of this layer for BP

Reimplemented from singa::Layer.

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/connection_layer.h

Public Member Functions

Additional Inherited Members

Detailed Description

Member Function Documentation