| allgather_precondition_gradient(lbann_comm &comm, ExeContextType &context) | lbann::KFAC | private |
| apply(ExecutionContext &context, model &m, data_coordinator &dc, execution_mode mode) final | lbann::KFAC | virtual |
| lbann::TrainingAlgorithm::apply(model &model, data_coordinator &dc) | lbann::TrainingAlgorithm | inline |
| damping_0_default | lbann::KFAC | static |
| damping_warmup_steps_default | lbann::KFAC | static |
| Device | lbann::KFAC | static |
| do_batch_begin_cbs(model &model) | lbann::KFAC | protected |
| do_batch_end_cbs(model &model) | lbann::KFAC | protected |
| do_epoch_begin_cbs(model &model) | lbann::KFAC | protected |
| do_epoch_end_cbs(model &model) | lbann::KFAC | protected |
| do_get_new_execution_context() const final | lbann::KFAC | protectedvirtual |
| do_train_begin_cbs(model &model) | lbann::KFAC | protected |
| do_train_end_cbs(model &model) | lbann::KFAC | protected |
| end_old_async_weights_model(model &model, lbann_comm *comm, ExeContextType &context) | lbann::KFAC | private |
| end_send_recv_inverse_matrices(ExeContextType &context, lbann_comm *comm) | lbann::KFAC | protected |
| end_sync_weights_async(model &model, lbann_comm *comm) | lbann::KFAC | private |
| ExeContextType typedef | lbann::KFAC | |
| get_name() const noexcept | lbann::TrainingAlgorithm | |
| get_new_execution_context() const | lbann::TrainingAlgorithm | inline |
| get_type() const final | lbann::KFAC | virtual |
| KFAC(std::string name, std::unique_ptr< TermCriteriaType > stop, std::vector< double > damping_act_params, std::vector< double > damping_err_params, std::vector< double > damping_bn_act_params, std::vector< double > damping_bn_err_params, std::vector< bool > kfac_use_interval, size_t damping_warmup_steps, double kronecker_decay, bool print_time, bool print_matrix, bool print_matrix_summary, bool use_pi, std::vector< size_t > update_intervals, size_t update_interval_steps, kfac::kfac_inverse_strategy inverse_strategy, std::vector< std::string > disable_layers, double learning_rate_factor, double learning_rate_factor_gru, size_t compute_interval, bool distribute_precondition_compute, bool use_eigen_decomposition, bool enable_copy_errors, bool enable_copy_activations) | lbann::KFAC | |
| KFAC(KFAC const &other)=delete | lbann::KFAC | |
| KFAC(KFAC &&other)=default | lbann::KFAC | |
| kronecker_decay_default | lbann::KFAC | static |
| m_compute_interval | lbann::KFAC | private |
| m_damping_act_params | lbann::KFAC | private |
| m_damping_bn_act_params | lbann::KFAC | private |
| m_damping_bn_err_params | lbann::KFAC | private |
| m_damping_err_params | lbann::KFAC | private |
| m_damping_warmup_steps | lbann::KFAC | private |
| m_disable_layers | lbann::KFAC | private |
| m_distribute_precondition_compute | lbann::KFAC | private |
| m_enable_copy_activations | lbann::KFAC | private |
| m_enable_copy_errors | lbann::KFAC | private |
| m_global_inverse_buffer_size | lbann::KFAC | private |
| m_has_kronecker_inverse | lbann::KFAC | private |
| m_inverse_matrices_size | lbann::KFAC | private |
| m_inverse_matrix_communication_reqs | lbann::KFAC | private |
| m_inverse_strategy | lbann::KFAC | private |
| m_kronecker_decay | lbann::KFAC | private |
| m_learning_rate_factor | lbann::KFAC | private |
| m_learning_rate_factor_gru | lbann::KFAC | private |
| m_print_matrix | lbann::KFAC | private |
| m_print_matrix_summary | lbann::KFAC | private |
| m_print_time | lbann::KFAC | private |
| m_stopping_criteria | lbann::KFAC | private |
| m_time_backward_pass | lbann::KFAC | private |
| m_time_forward_pass | lbann::KFAC | private |
| m_time_kfac | lbann::KFAC | private |
| m_time_span_backward_comm | lbann::KFAC | private |
| m_time_span_backward_comm_end | lbann::KFAC | private |
| m_time_span_forward_comm | lbann::KFAC | private |
| m_time_span_forward_comm_end | lbann::KFAC | private |
| m_time_span_inverse_comm | lbann::KFAC | private |
| m_time_span_inverse_send_recv | lbann::KFAC | private |
| m_time_span_precond_comm | lbann::KFAC | private |
| m_update_interval_steps | lbann::KFAC | private |
| m_update_intervals | lbann::KFAC | private |
| m_use_eigen_decomposition | lbann::KFAC | private |
| m_use_KFAC_epoch | lbann::KFAC | private |
| m_use_pi | lbann::KFAC | private |
| m_weight_matrices_buffer_size | lbann::KFAC | private |
| m_weights_communication_reqs | lbann::KFAC | private |
| on_backward_prop_end(ExeContextType &context, model &model) | lbann::KFAC | private |
| on_forward_prop_end(ExeContextType &context, model &model) | lbann::KFAC | private |
| operator=(const KFAC &other)=delete | lbann::KFAC | |
| operator=(KFAC &&other)=default | lbann::KFAC | |
| lbann::TrainingAlgorithm::operator=(const TrainingAlgorithm &other)=delete | lbann::TrainingAlgorithm | protected |
| lbann::TrainingAlgorithm::operator=(TrainingAlgorithm &&other)=default | lbann::TrainingAlgorithm | protected |
| prof_color | lbann::KFAC | static |
| prof_sync | lbann::KFAC | static |
| setup_models(std::vector< observer_ptr< model >> const &models, size_t max_mini_batch_size, const std::vector< El::Grid *> &grids) | lbann::TrainingAlgorithm | |
| start_old_async_weights_model(model &model, lbann_comm *comm, ExeContextType &context) | lbann::KFAC | private |
| start_send_recv_inverse_matrices(ExeContextType &context, lbann_comm *comm) | lbann::KFAC | protected |
| start_sync_weights_async(model &model, lbann_comm *comm) | lbann::KFAC | private |
| sync_weights_model(model &model, lbann_comm *comm) | lbann::KFAC | private |
| TermCriteriaType typedef | lbann::KFAC | |
| train(ExeContextType &c, model &model, data_coordinator &dc, TermCriteriaType const &term) | lbann::KFAC | |
| train_mini_batch(ExeContextType &c, model &model, data_coordinator &dc) | lbann::KFAC | protected |
| TrainingAlgorithm(std::string name) | lbann::TrainingAlgorithm | |
| TrainingAlgorithm(const TrainingAlgorithm &other)=delete | lbann::TrainingAlgorithm | protected |
| TrainingAlgorithm(TrainingAlgorithm &&other)=default | lbann::TrainingAlgorithm | protected |
| ~KFAC() noexcept=default | lbann::KFAC | |
| ~TrainingAlgorithm()=default | lbann::TrainingAlgorithm | virtual |