lbann::optimizer::GradientHelperImpl< TensorDataType > Class Template Reference

#include <optimizer.hpp>

Inheritance diagram for lbann::optimizer::GradientHelperImpl< TensorDataType >:

Collaboration diagram for lbann::optimizer::GradientHelperImpl< TensorDataType >:

Public Types
using	AbsDistMatType = El::AbstractDistMatrix< TensorDataType >

Public Member Functions
	GradientHelperImpl (El::Int height, El::Int width, El::DistData dist_data)
AbsDistMatType &	gradient () noexcept override
AbsDistMatType const &	gradient () const noexcept override
void	start_allreduce (lbann_comm &comm) override
void	complete_allreduce (lbann_comm &comm) override
void	clear () override
Public Member Functions inherited from lbann::optimizer::GradientHelper
virtual	~GradientHelper ()=default
optimizer_gradient_status	get_status () const noexcept
void	set_status (optimizer_gradient_status s) noexcept

Private Attributes
std::unique_ptr< AbsDistMatType >	gradient_
Al::request	allreduce_req_

Detailed Description

template<typename TensorDataType>
class lbann::optimizer::GradientHelperImpl< TensorDataType >

Definition at line 233 of file optimizer.hpp.

Member Typedef Documentation

AbsDistMatType

template<typename TensorDataType >

using lbann::optimizer::GradientHelperImpl< TensorDataType >::AbsDistMatType = El::AbstractDistMatrix<TensorDataType>

Definition at line 236 of file optimizer.hpp.

Constructor & Destructor Documentation

GradientHelperImpl()

template<typename TensorDataType >

lbann::optimizer::GradientHelperImpl< TensorDataType >::GradientHelperImpl ( El::Int  height, El::Int  width, El::DistData  dist_data  )

inline

Definition at line 239 of file optimizer.hpp.

Member Function Documentation

clear()

template<typename TensorDataType >

void lbann::optimizer::GradientHelperImpl< TensorDataType >::clear ( )

overridevirtual

Implements lbann::optimizer::GradientHelper.

Definition at line 233 of file optimizer_impl.hpp.

complete_allreduce()

template<typename TensorDataType >

void lbann::optimizer::GradientHelperImpl< TensorDataType >::complete_allreduce ( lbann_comm &  comm )

overridevirtual

Implements lbann::optimizer::GradientHelper.

Definition at line 210 of file optimizer_impl.hpp.

Here is the call graph for this function:

gradient() [1/2]

template<typename TensorDataType >

AbsDistMatType& lbann::optimizer::GradientHelperImpl< TensorDataType >::gradient ( )

inlineoverridevirtualnoexcept

Implements lbann::optimizer::GradientHelper.

Definition at line 244 of file optimizer.hpp.

gradient() [2/2]

template<typename TensorDataType >

AbsDistMatType const& lbann::optimizer::GradientHelperImpl< TensorDataType >::gradient ( ) const

inlineoverridevirtualnoexcept

Implements lbann::optimizer::GradientHelper.

Definition at line 245 of file optimizer.hpp.

start_allreduce()

template<typename TensorDataType >

void lbann::optimizer::GradientHelperImpl< TensorDataType >::start_allreduce ( lbann_comm &  comm )

overridevirtual

Implements lbann::optimizer::GradientHelper.

Definition at line 190 of file optimizer_impl.hpp.

Here is the call graph for this function:

Member Data Documentation

allreduce_req_

template<typename TensorDataType >

Al::request lbann::optimizer::GradientHelperImpl< TensorDataType >::allreduce_req_

private

Definition at line 255 of file optimizer.hpp.

gradient_

template<typename TensorDataType >

std::unique_ptr<AbsDistMatType> lbann::optimizer::GradientHelperImpl< TensorDataType >::gradient_

private

Definition at line 254 of file optimizer.hpp.

---

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

• optimizer.hpp
• optimizer_impl.hpp