| compute_local_kronecker_factors(lbann_comm *comm, bool print_matrix, bool print_matrix_summary) override | lbann::kfac_block_fc_conv< Device > | virtual |
| compute_pi(const El::Matrix< DataType, Device > &A, const El::Matrix< DataType, Device > &G, El::Matrix< DataType, Device > &ws, const El::SyncInfo< Device > &sync_info) | lbann::kfac_block_fc_conv< Device > | privatestatic |
| compute_preconditioned_gradients(lbann_comm *comm, DataType learning_rate_factor, bool print_matrix, bool print_matrix_summary, bool print_time) override | lbann::kfac_block_fc_conv< Device > | virtual |
| end_communication_backward_end(lbann_comm *comm) override | lbann::kfac_block_fc_conv< Device > | virtual |
| end_communication_forward_end(lbann_comm *comm) override | lbann::kfac_block_fc_conv< Device > | virtual |
| get_conv_layer() | lbann::kfac_block_fc_conv< Device > | inlineprivate |
| get_current_batch_size() | lbann::kfac_block< Device > | inline |
| get_gradient_wrt_weight_buffer(int index) | lbann::kfac_block< Device > | inline |
| get_info() const override | lbann::kfac_block_fc_conv< Device > | inlinevirtual |
| get_input_size() | lbann::kfac_block< Device > | inline |
| get_internal_matrix_info() const override | lbann::kfac_block_fc_conv< Device > | privatevirtual |
| get_inverse_matrices(El::Matrix< DataType, Device > &output, int offset) override | lbann::kfac_block_fc_conv< Device > | virtual |
| get_inverse_matrices_size(lbann_comm *comm) override | lbann::kfac_block_fc_conv< Device > | virtual |
| get_inverse_matrices_size_vector(lbann_comm *comm) override | lbann::kfac_block_fc_conv< Device > | virtual |
| get_inverse_proc_rank() const | lbann::kfac_block< Device > | inline |
| get_kronecker_factor_conv(El::Matrix< DataType, Device > &factor, El::Matrix< DataType, Device > &Acol, const El::Matrix< DataType, Device > &activations, DataType alpha, size_t local_batch_size, size_t num_channels, const std::vector< int > &spatial_dims, const convolution_layer< DataType, data_layout::DATA_PARALLEL, Device > *l_conv, bool use_im2col, const El::SyncInfo< Device > &sync_info) | lbann::kfac_block_fc_conv< Device > | privatestatic |
| get_kronecker_factor_fc(El::AbstractMatrix< DataType > &factor, const El::AbstractMatrix< DataType > &activations, DataType alpha) | lbann::kfac_block_fc_conv< Device > | privatestatic |
| get_local_activation_buffer(int index) | lbann::kfac_block< Device > | inline |
| get_local_error_buffer(int index) | lbann::kfac_block< Device > | inline |
| get_local_kronecker_buffers() override | lbann::kfac_block_fc_conv< Device > | inlinevirtual |
| get_local_memory_consumption() override | lbann::kfac_block_fc_conv< Device > | inlinevirtual |
| get_name() const | lbann::kfac_block< Device > | inline |
| get_output_size() | lbann::kfac_block< Device > | inline |
| get_preconditioned_grad_buffers() override | lbann::kfac_block_fc_conv< Device > | virtual |
| get_sync_info() | lbann::kfac_block< Device > | protected |
| get_weight_buffer(int index) | lbann::kfac_block< Device > | inline |
| get_workspace_matrix(const std::string &key, size_t height, size_t width) | lbann::kfac_block< Device > | protected |
| initialize_activations_and_errors(lbann_comm *comm, int num_local_activations, int num_local_errors, int num_weights) override | lbann::kfac_block_fc_conv< Device > | virtual |
| kfac_block(Layer *layer, kfac::KFACExecutionContext *context, size_t layer_id, size_t inverse_proc_rank, bool enable_copy_errors, bool enable_copy_activations, int input_size, int output_size) | lbann::kfac_block< Device > | inline |
| kfac_block_fc_conv(Layer *layer, kfac::KFACExecutionContext *context, const size_t layer_id, const size_t inverse_proc_rank, const bool enable_copy_errors, const bool enable_copy_activations, const int input_size, const int output_size, const bool is_conv) | lbann::kfac_block_fc_conv< Device > | inline |
| kfac_block_fc_conv(const kfac_block_fc_conv &)=default | lbann::kfac_block_fc_conv< Device > | |
| m_activations_copy | lbann::kfac_block< Device > | protected |
| m_Ainv_height | lbann::kfac_block_fc_conv< Device > | private |
| m_Ainv_width | lbann::kfac_block_fc_conv< Device > | private |
| m_batch_size | lbann::kfac_block< Device > | protected |
| m_child_local_errors | lbann::kfac_block< Device > | protected |
| m_conv_input_spatial_dims | lbann::kfac_block_fc_conv< Device > | private |
| m_conv_input_spatial_prod | lbann::kfac_block_fc_conv< Device > | private |
| m_conv_output_spatial_dims | lbann::kfac_block_fc_conv< Device > | private |
| m_conv_output_spatial_prod | lbann::kfac_block_fc_conv< Device > | private |
| m_enable_copy_activations | lbann::kfac_block< Device > | protected |
| m_enable_copy_errors | lbann::kfac_block< Device > | protected |
| m_errors_copy | lbann::kfac_block< Device > | protected |
| m_Ginv_height | lbann::kfac_block_fc_conv< Device > | private |
| m_Ginv_width | lbann::kfac_block_fc_conv< Device > | private |
| m_grad_buffer_v | lbann::kfac_block_fc_conv< Device > | private |
| m_has_bias | lbann::kfac_block_fc_conv< Device > | private |
| m_has_kronecker_inverse | lbann::kfac_block< Device > | protected |
| m_height_A | lbann::kfac_block_fc_conv< Device > | private |
| m_height_G | lbann::kfac_block_fc_conv< Device > | private |
| m_input_size | lbann::kfac_block< Device > | protected |
| m_inverse_proc_rank | lbann::kfac_block< Device > | protected |
| m_is_conv | lbann::kfac_block_fc_conv< Device > | private |
| m_kronecker_average_A | lbann::kfac_block_fc_conv< Device > | private |
| m_kronecker_average_G | lbann::kfac_block_fc_conv< Device > | private |
| m_kronecker_factor_buf_A | lbann::kfac_block_fc_conv< Device > | private |
| m_kronecker_factor_buf_G | lbann::kfac_block_fc_conv< Device > | private |
| m_kronecker_inverse_A | lbann::kfac_block_fc_conv< Device > | private |
| m_kronecker_inverse_G | lbann::kfac_block_fc_conv< Device > | private |
| m_layer | lbann::kfac_block< Device > | protected |
| m_layer_id | lbann::kfac_block< Device > | protected |
| m_output_size | lbann::kfac_block< Device > | protected |
| m_parent_local_activations | lbann::kfac_block< Device > | protected |
| m_requests_backward_end | lbann::kfac_block< Device > | protected |
| m_requests_forward_end | lbann::kfac_block< Device > | protected |
| m_subset_matrix | lbann::kfac_block< Device > | protected |
| m_weight_gradients | lbann::kfac_block< Device > | protected |
| m_weight_values | lbann::kfac_block< Device > | protected |
| on_forward_prop_end(lbann_comm *comm) | lbann::kfac_block< Device > | inlinevirtual |
| operator=(const kfac_block_fc_conv &)=default | lbann::kfac_block_fc_conv< Device > | |
| resize_inverse_matrices_size(El::Matrix< double, El::Device::CPU > &inverse_matrices_size, int block_number) override | lbann::kfac_block_fc_conv< Device > | virtual |
| set_current_batch_size(El::Int batch_size) | lbann::kfac_block< Device > | inline |
| set_inverse_matrices(El::Matrix< DataType, Device > &workspace, int offset, lbann_comm *comm) override | lbann::kfac_block_fc_conv< Device > | virtual |
| start_communication_backward_end(lbann_comm *comm) override | lbann::kfac_block_fc_conv< Device > | virtual |
| start_communication_forward_end(lbann_comm *comm) override | lbann::kfac_block_fc_conv< Device > | virtual |
| update_kronecker_average(lbann_comm *comm, DataType kronecker_decay, bool print_matrix, bool print_matrix_summary) override | lbann::kfac_block_fc_conv< Device > | virtual |
| update_kronecker_inverse(lbann_comm *comm, bool use_pi, DataType damping_act, DataType damping_err, DataType learning_rate_factor, bool use_eigen_decomposition, bool print_matrix, bool print_matrix_summary, bool print_time) override | lbann::kfac_block_fc_conv< Device > | virtual |
| ~kfac_block()=default | lbann::kfac_block< Device > | virtual |
1.8.13