| check_dnn_lib_spec() const | lbann::kfac_block_gru< Device > | private |
| compute_local_kronecker_factors(lbann_comm *comm, bool print_matrix, bool print_matrix_summary) override | lbann::kfac_block_gru< Device > | virtual |
| compute_preconditioned_gradients(lbann_comm *comm, DataType learning_rate_factor, bool print_matrix, bool print_matrix_summary, bool print_time) override | lbann::kfac_block_gru< Device > | virtual |
| end_communication_backward_end(lbann_comm *comm) override | lbann::kfac_block_gru< Device > | virtual |
| end_communication_forward_end(lbann_comm *comm) override | lbann::kfac_block_gru< Device > | virtual |
| get_current_batch_size() | lbann::kfac_block< Device > | inline |
| get_gradient_buffer(kfac_gru_util::weight_type matrix_type, El::Matrix< DataType, Device > &view) | lbann::kfac_block_gru< Device > | private |
| get_gradient_matrix(kfac_gru_util::weight_type matrix_type, El::Matrix< DataType, Device > &view) | lbann::kfac_block_gru< Device > | private |
| get_gradient_wrt_weight_buffer(int index) | lbann::kfac_block< Device > | inline |
| get_gru_layer() const | lbann::kfac_block_gru< Device > | inlineprivate |
| get_hidden_size() const | lbann::kfac_block_gru< Device > | inlineprivate |
| get_info() const | lbann::kfac_block< Device > | inlinevirtual |
| get_input_size() const | lbann::kfac_block_gru< Device > | inlineprivate |
| lbann::kfac_block::get_input_size() | lbann::kfac_block< Device > | inline |
| get_internal_matrix_info() const override | lbann::kfac_block_gru< Device > | privatevirtual |
| get_inverse_matrices(El::Matrix< DataType, Device > &output, int offset) override | lbann::kfac_block_gru< Device > | virtual |
| get_inverse_matrices_size(lbann_comm *comm) override | lbann::kfac_block_gru< Device > | virtual |
| get_inverse_matrices_size_vector(lbann_comm *comm) override | lbann::kfac_block_gru< Device > | inlinevirtual |
| get_inverse_proc_rank() const | lbann::kfac_block< Device > | inline |
| 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_gru< Device > | virtual |
| get_local_memory_consumption() override | lbann::kfac_block_gru< 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_gru< Device > | virtual |
| get_r_i(El::Matrix< DataType, Device > &r, El::Matrix< DataType, Device > &i, const El::Matrix< DataType, Device > &biases_ones, const El::Matrix< DataType, Device > &local_inputs, const El::Matrix< DataType, Device > &local_outputs, const El::Matrix< DataType, Device > &h0, size_t local_batch_size, const El::SyncInfo< Device > &sync_info) | lbann::kfac_block_gru< Device > | private |
| get_seq_length() const | lbann::kfac_block_gru< Device > | inlineprivate |
| get_sync_info() | lbann::kfac_block< Device > | protected |
| get_weight_buffer(int index) | lbann::kfac_block< Device > | inline |
| get_weight_matrix(kfac_gru_util::weight_type matrix_type, El::Matrix< DataType, Device > &view) | lbann::kfac_block_gru< Device > | private |
| 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_gru< 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_gru(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_gru< Device > | inline |
| kfac_block_gru(const kfac_block_gru &)=default | lbann::kfac_block_gru< Device > | |
| m_activations_copy | lbann::kfac_block< Device > | protected |
| m_batch_size | lbann::kfac_block< Device > | protected |
| m_child_local_errors | lbann::kfac_block< Device > | protected |
| 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_grad_buffer_A_h | lbann::kfac_block_gru< Device > | private |
| m_grad_buffer_A_x | lbann::kfac_block_gru< Device > | private |
| m_grad_buffer_G | lbann::kfac_block_gru< Device > | private |
| m_has_kronecker_inverse | lbann::kfac_block< Device > | protected |
| m_input_size | lbann::kfac_block< Device > | protected |
| m_inverse_proc_rank | lbann::kfac_block< Device > | protected |
| m_kronecker_average_A_h | lbann::kfac_block_gru< Device > | private |
| m_kronecker_average_A_x | lbann::kfac_block_gru< Device > | private |
| m_kronecker_average_G | lbann::kfac_block_gru< Device > | private |
| m_kronecker_factor_buf_A_h | lbann::kfac_block_gru< Device > | private |
| m_kronecker_factor_buf_A_x | lbann::kfac_block_gru< Device > | private |
| m_kronecker_factor_buf_G | lbann::kfac_block_gru< Device > | private |
| m_kronecker_inverse_A_h | lbann::kfac_block_gru< Device > | private |
| m_kronecker_inverse_A_x | lbann::kfac_block_gru< Device > | private |
| m_kronecker_inverse_G | lbann::kfac_block_gru< 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_requests_workspace | lbann::kfac_block_gru< Device > | private |
| m_reserve_space_fwd | lbann::kfac_block_gru< Device > | private |
| m_reserve_space_fwd_size | lbann::kfac_block_gru< Device > | private |
| 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) override | lbann::kfac_block_gru< Device > | virtual |
| operator=(const kfac_block_gru &)=default | lbann::kfac_block_gru< Device > | |
| resize_inverse_matrices_size(El::Matrix< double, El::Device::CPU > &inverse_matrices_size, int block_number) override | lbann::kfac_block_gru< Device > | inlinevirtual |
| send_recv_reserve_space(lbann_comm *comm) | lbann::kfac_block_gru< Device > | private |
| 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_gru< Device > | virtual |
| start_communication_backward_end(lbann_comm *comm) override | lbann::kfac_block_gru< Device > | virtual |
| start_communication_forward_end(lbann_comm *comm) override | lbann::kfac_block_gru< Device > | virtual |
| update_kronecker_average(lbann_comm *comm, DataType kronecker_decay, bool print_matrix, bool print_matrix_summary) override | lbann::kfac_block_gru< 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_gru< Device > | virtual |
| ~kfac_block()=default | lbann::kfac_block< Device > | virtual |
1.8.13