lbann::distconv_adapter Class Reference

#include <distconv_adapter.hpp>

Inheritance diagram for lbann::distconv_adapter:

Collaboration diagram for lbann::distconv_adapter:

Public Member Functions
	distconv_adapter (Layer &layer)
virtual	~distconv_adapter ()=default
virtual const dc::AbsTensor &	get_activations (const Layer &child) const =0
virtual const dc::AbsTensor &	get_error_signals (const Layer &parent) const =0
virtual void	setup_distributions (tensor_overlap_constraints &constraints)
void	impose_adjacent_overlap_constraints (tensor_overlap_constraints &constraints)
dc::Dist &	get_prev_activations_dist ()
const dc::Dist &	get_prev_activations_dist () const
dc::Dist &	get_activations_dist ()
const dc::Dist &	get_activations_dist () const
dc::Dist &	get_prev_error_signals_dist ()
const dc::Dist &	get_prev_error_signals_dist () const
dc::Dist &	get_error_signals_dist ()
const dc::Dist &	get_error_signals_dist () const
virtual void	setup_fp_tensors ()
virtual void	setup_bp_tensors ()
virtual void	setup_layer (size_t workspace_capacity)
virtual void	fp_setup ()=0
virtual void	fp_postprocess ()=0
virtual void	bp_setup ()=0
virtual void	bp_postprocess ()=0
virtual bool	parent_copy_required (size_t input_index) const
virtual bool	parent_shuffle_required (size_t input_index) const
virtual bool	child_copy_required (size_t output_index) const
virtual bool	child_shuffle_required (size_t output_index) const
virtual void	dump_activations () const =0
virtual void	dump_original_activations ()=0
virtual void	dump_error_signals () const =0
virtual void	dump_original_error_signals ()=0

Protected Member Functions
virtual Layer &	layer ()
virtual const Layer &	layer () const
std::string	get_name () const
virtual void	setup_prev_activations ()=0
virtual void	setup_original_prev_activations ()=0
virtual void	setup_activations ()=0
virtual void	setup_original_activations ()=0
virtual void	setup_prev_error_signals ()=0
virtual void	setup_original_prev_error_signals ()=0
virtual void	setup_error_signals ()=0
virtual void	setup_original_error_signals ()=0
virtual void	ensure_prev_activations ()=0
virtual void	copy_out_activations ()=0
virtual void	ensure_prev_error_signals ()=0
virtual void	copy_out_error_signals ()=0

Protected Attributes
std::vector< dc::Dist >	m_prev_activations_dists
std::vector< dc::Dist >	m_activations_dists
std::vector< dc::Dist >	m_prev_error_signals_dists
std::vector< dc::Dist >	m_error_signals_dists

Private Member Functions
void	setup_tensor_shuffle ()
void	adjust_parallel_strategy ()

Private Attributes
Layer &	m_layer
std::vector< bool >	m_parent_copy_required
std::vector< bool >	m_parent_shuffle_required
std::vector< bool >	m_child_copy_required
std::vector< bool >	m_child_shuffle_required

Friends
class	Layer

Detailed Description

Definition at line 69 of file distconv_adapter.hpp.

Constructor & Destructor Documentation

distconv_adapter()

lbann::distconv_adapter::distconv_adapter

(

Layer &

layer

)

~distconv_adapter()

virtual lbann::distconv_adapter::~distconv_adapter ( )

virtualdefault

Member Function Documentation

adjust_parallel_strategy()

void lbann::distconv_adapter::adjust_parallel_strategy ( )

private

bp_postprocess()

virtual void lbann::distconv_adapter::bp_postprocess ( )

pure virtual

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

bp_setup()

virtual void lbann::distconv_adapter::bp_setup ( )

pure virtual

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

child_copy_required()

virtual bool lbann::distconv_adapter::child_copy_required ( size_t  output_index ) const

virtual

child_shuffle_required()

virtual bool lbann::distconv_adapter::child_shuffle_required ( size_t  output_index ) const

virtual

copy_out_activations()

virtual void lbann::distconv_adapter::copy_out_activations ( )

protectedpure virtual

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

copy_out_error_signals()

virtual void lbann::distconv_adapter::copy_out_error_signals ( )

protectedpure virtual

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

dump_activations()

virtual void lbann::distconv_adapter::dump_activations ( ) const

pure virtual

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

dump_error_signals()

virtual void lbann::distconv_adapter::dump_error_signals ( ) const

pure virtual

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

dump_original_activations()

virtual void lbann::distconv_adapter::dump_original_activations ( )

pure virtual

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

dump_original_error_signals()

virtual void lbann::distconv_adapter::dump_original_error_signals ( )

pure virtual

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

ensure_prev_activations()

virtual void lbann::distconv_adapter::ensure_prev_activations ( )

protectedpure virtual

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

ensure_prev_error_signals()

virtual void lbann::distconv_adapter::ensure_prev_error_signals ( )

protectedpure virtual

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

fp_postprocess()

virtual void lbann::distconv_adapter::fp_postprocess ( )

pure virtual

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

fp_setup()

virtual void lbann::distconv_adapter::fp_setup ( )

pure virtual

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

get_activations()

virtual const dc::AbsTensor& lbann::distconv_adapter::get_activations ( const Layer &  child ) const

pure virtual

Get activation tensor corresponding to child layer.

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

get_activations_dist() [1/2]

dc::Dist& lbann::distconv_adapter::get_activations_dist

(

)

get_activations_dist() [2/2]

const dc::Dist& lbann::distconv_adapter::get_activations_dist

(

)

const

get_error_signals()

virtual const dc::AbsTensor& lbann::distconv_adapter::get_error_signals ( const Layer &  parent ) const

pure virtual

Get error signal tensor corresponding to parent layer.

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

get_error_signals_dist() [1/2]

dc::Dist& lbann::distconv_adapter::get_error_signals_dist

(

)

get_error_signals_dist() [2/2]

const dc::Dist& lbann::distconv_adapter::get_error_signals_dist

(

)

const

get_name()

std::string lbann::distconv_adapter::get_name ( ) const

protected

get_prev_activations_dist() [1/2]

dc::Dist& lbann::distconv_adapter::get_prev_activations_dist

(

)

get_prev_activations_dist() [2/2]

const dc::Dist& lbann::distconv_adapter::get_prev_activations_dist

(

)

const

get_prev_error_signals_dist() [1/2]

dc::Dist& lbann::distconv_adapter::get_prev_error_signals_dist

(

)

get_prev_error_signals_dist() [2/2]

const dc::Dist& lbann::distconv_adapter::get_prev_error_signals_dist

(

)

const

impose_adjacent_overlap_constraints()

void lbann::distconv_adapter::impose_adjacent_overlap_constraints

(

tensor_overlap_constraints &

constraints

)

layer() [1/2]

virtual Layer& lbann::distconv_adapter::layer ( )

protectedvirtual

layer() [2/2]

virtual const Layer& lbann::distconv_adapter::layer ( ) const

protectedvirtual

parent_copy_required()

virtual bool lbann::distconv_adapter::parent_copy_required ( size_t  input_index ) const

virtual

parent_shuffle_required()

virtual bool lbann::distconv_adapter::parent_shuffle_required ( size_t  input_index ) const

virtual

setup_activations()

virtual void lbann::distconv_adapter::setup_activations ( )

protectedpure virtual

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

setup_bp_tensors()

virtual void lbann::distconv_adapter::setup_bp_tensors ( )

virtual

Here is the caller graph for this function:

setup_distributions()

virtual void lbann::distconv_adapter::setup_distributions ( tensor_overlap_constraints &  constraints )

virtual

Here is the caller graph for this function:

setup_error_signals()

virtual void lbann::distconv_adapter::setup_error_signals ( )

protectedpure virtual

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

setup_fp_tensors()

virtual void lbann::distconv_adapter::setup_fp_tensors ( )

virtual

Here is the caller graph for this function:

setup_layer()

virtual void lbann::distconv_adapter::setup_layer ( size_t  workspace_capacity )

inlinevirtual

Definition at line 98 of file distconv_adapter.hpp.

Here is the caller graph for this function:

setup_original_activations()

virtual void lbann::distconv_adapter::setup_original_activations ( )

protectedpure virtual

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

setup_original_error_signals()

virtual void lbann::distconv_adapter::setup_original_error_signals ( )

protectedpure virtual

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

setup_original_prev_activations()

virtual void lbann::distconv_adapter::setup_original_prev_activations ( )

protectedpure virtual

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

setup_original_prev_error_signals()

virtual void lbann::distconv_adapter::setup_original_prev_error_signals ( )

protectedpure virtual

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

setup_prev_activations()

virtual void lbann::distconv_adapter::setup_prev_activations ( )

protectedpure virtual

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

setup_prev_error_signals()

virtual void lbann::distconv_adapter::setup_prev_error_signals ( )

protectedpure virtual

Implemented in lbann::data_type_distconv_adapter< InputTensorDataType, OutputTensorDataType >.

setup_tensor_shuffle()

void lbann::distconv_adapter::setup_tensor_shuffle ( )

private

Friends And Related Function Documentation

Layer

friend class Layer

friend

Definition at line 71 of file distconv_adapter.hpp.

Member Data Documentation

m_activations_dists

std::vector<dc::Dist> lbann::distconv_adapter::m_activations_dists

protected

Definition at line 136 of file distconv_adapter.hpp.

m_child_copy_required

std::vector<bool> lbann::distconv_adapter::m_child_copy_required

private

Definition at line 144 of file distconv_adapter.hpp.

m_child_shuffle_required

std::vector<bool> lbann::distconv_adapter::m_child_shuffle_required

private

Definition at line 145 of file distconv_adapter.hpp.

m_error_signals_dists

std::vector<dc::Dist> lbann::distconv_adapter::m_error_signals_dists

protected

Definition at line 138 of file distconv_adapter.hpp.

m_layer

Layer& lbann::distconv_adapter::m_layer

private

Definition at line 141 of file distconv_adapter.hpp.

m_parent_copy_required

std::vector<bool> lbann::distconv_adapter::m_parent_copy_required

private

Definition at line 142 of file distconv_adapter.hpp.

m_parent_shuffle_required

std::vector<bool> lbann::distconv_adapter::m_parent_shuffle_required

private

Definition at line 143 of file distconv_adapter.hpp.

m_prev_activations_dists

std::vector<dc::Dist> lbann::distconv_adapter::m_prev_activations_dists

protected

Definition at line 135 of file distconv_adapter.hpp.

m_prev_error_signals_dists

std::vector<dc::Dist> lbann::distconv_adapter::m_prev_error_signals_dists

protected

Definition at line 137 of file distconv_adapter.hpp.

---

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

• distconv_adapter.hpp