LBANN
0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
lbann::optimizer Member List
This is the complete list of members for
lbann::optimizer
, including all inherited members.
accumulate_all_gradient_contributions
(El::AbstractDistMatrix< TensorDataType > &gradient)
lbann::optimizer
add_gradient_source
(const void *source)
lbann::optimizer
add_to_gradient
(El::AbstractDistMatrix< TensorDataType > const &contrib, TensorDataType scale=1.f, bool allreduce_needed=false)
lbann::optimizer
clear_gradient
()
lbann::optimizer
clone
() const
lbann::Cloneable< HasAbstractFunction< optimizer > >
inline
finish_gradient_allreduce
()
lbann::optimizer
get_comm
()
lbann::optimizer
inline
get_comm
() const
lbann::optimizer
inline
get_description
() const
lbann::optimizer
virtual
get_gradient_buffer
(TensorDataType &buf_scale, TensorDataType &in_scale, bool allreduce_needed=false)
lbann::optimizer
get_gradient_sources
()
lbann::optimizer
inline
get_gradient_status
() const
lbann::optimizer
inline
get_learning_rate
() const =0
lbann::optimizer
pure virtual
get_matrix_info
() const =0
lbann::optimizer
pure virtual
get_num_gradient_sources
() const
lbann::optimizer
get_step_time
() const
lbann::optimizer
inline
get_type
() const =0
lbann::optimizer
pure virtual
gradient_manager_ptr
typedef
lbann::optimizer
private
gradient_manager_type
typedef
lbann::optimizer
private
gradients_
lbann::optimizer
private
inc_step_time
(EvalType time)
lbann::optimizer
inline
m_comm
lbann::optimizer
private
m_gradient_sources
lbann::optimizer
private
m_gradient_status
lbann::optimizer
private
m_step_time
lbann::optimizer
private
operator=
(const optimizer &other)
lbann::optimizer
optimizer
()
lbann::optimizer
optimizer
(const optimizer &other)
lbann::optimizer
remove_gradient_source
(const void *source)
lbann::optimizer
reset_counters
()
lbann::optimizer
inline
virtual
serialize
(Archive &ar)
lbann::optimizer
set_comm
(lbann_comm &comm)
lbann::optimizer
inline
set_gradient_status
(const optimizer_gradient_status status)
lbann::optimizer
inline
set_learning_rate
(double)=0
lbann::optimizer
pure virtual
set_step_time
(EvalType time)
lbann::optimizer
inline
setup
(weights *w)=0
lbann::optimizer
pure virtual
start_gradient_allreduce
()
lbann::optimizer
step
()=0
lbann::optimizer
pure virtual
write_proto
(lbann_data::Optimizer &proto) const =0
lbann::optimizer
pure virtual
~optimizer
()=default
lbann::optimizer
virtual
Generated on Wed Oct 11 2023 20:49:40 for LBANN by
1.8.13