lbann::model Member List

This is the complete list of members for lbann::model, including all inherited members.

add_callback(std::shared_ptr< callback_base > cb)	lbann::model
add_dummy_layers(std::unordered_set< std::string > &layer_names)	lbann::model	private
add_evaluation_layers(std::unordered_set< Layer *> &layer_set, std::unordered_set< std::string > &layer_names)	lbann::model	private
add_layer(OwningLayerPtr &&l)	lbann::model
add_metric(std::unique_ptr< metric > m)	lbann::model
add_split_layers(std::unordered_set< std::string > &layer_names)	lbann::model	private
add_weights(OwningWeightsPtr &&w)	lbann::model
allow_background_io_activity(bool enable) noexcept	lbann::model	inline
apply_subgraph_parallelism	lbann::model	private
background_io_activity_allowed() const noexcept	lbann::model	inline
backward_prop(bool compute_weight_grads_only=true)	lbann::model
check_subgraph_parallelism()	lbann::model	private
clear_gradients()	lbann::model
copy_trained_weights_from(std::vector< weights *> &w)	lbann::model
create_optimizer() const	lbann::model	inline
do_layer_backward_prop_begin_cbs(Layer *l)	lbann::model
do_layer_backward_prop_end_cbs(Layer *l)	lbann::model
do_layer_forward_prop_begin_cbs(execution_mode mode, Layer *l)	lbann::model
do_layer_forward_prop_end_cbs(execution_mode mode, Layer *l)	lbann::model
do_model_backward_prop_begin_cbs()	lbann::model
do_model_backward_prop_end_cbs()	lbann::model
do_model_forward_prop_begin_cbs(execution_mode mode)	lbann::model
do_model_forward_prop_end_cbs(execution_mode mode)	lbann::model
do_model_optimize_begin_cbs()	lbann::model
do_model_optimize_end_cbs()	lbann::model
do_setup_end_cbs()	lbann::model
do_weight_optimize_begin_cbs(weights *w)	lbann::model
do_weight_optimize_end_cbs(weights *w)	lbann::model
enable_subgraph_parallelism() noexcept	lbann::model	inline
enable_subgraph_topology	lbann::model	private
ensure_input_layers_first()	lbann::model	private
evaluate_metrics(execution_mode mode, size_t current_mini_batch_size)	lbann::model
forward_prop(execution_mode mode)	lbann::model
get_callbacks()	lbann::model	inline
get_callbacks_with_ownership() noexcept	lbann::model	inline
get_comm() const noexcept	lbann::model	inline
get_current_mini_batch_size() const noexcept	lbann::model	inline
get_description() const	lbann::model
get_execution_context() const	lbann::model	inline
get_execution_context()	lbann::model	inline
get_layer(El::Int pos)	lbann::model
get_layer(El::Int pos) const	lbann::model
get_layers()	lbann::model
get_layers() const	lbann::model
get_max_mini_batch_size() const noexcept	lbann::model	inline
get_max_subgraph_branches()	lbann::model	private
get_metrics()	lbann::model
get_metrics() const	lbann::model
get_name() const noexcept	lbann::model	inline
get_num_layers() const noexcept	lbann::model
get_num_resources_branch_layers() const noexcept	lbann::model	inline
get_num_resources_non_branch_layers() const noexcept	lbann::model	inline
get_objective_function() const noexcept	lbann::model	inline
get_objective_function() noexcept	lbann::model	inline
get_parent_subgrid_tags(int layer_index)	lbann::model	private
get_resources_for_input_layer(std::vector< int > &masterSubGrid, int num_subgrids)	lbann::model	private
get_resources_for_merge_layers(std::set< int > &pooled_set, int child_index, int num_subgrids)	lbann::model	private
get_resources_for_spliting_point(std::vector< int > &parent_ranks, std::vector< int > &subgrid_ranks, int layer_index, int number_ranks_in_grid, int num_subgrids)	lbann::model	private
get_subgraph_num_parent_resources() const noexcept	lbann::model	inline
get_subgraph_subgrids_ranks(std::vector< int > &parent_ranks, std::vector< int > &subgrid_ranks, int layer_index, int number_ranks_in_grid)	lbann::model	private
get_subgrid_communication_type() const noexcept	lbann::model	inline
get_subgrid_topology() const noexcept	lbann::model	inline
get_subgrids_order(std::vector< int > &ranks_order, int num_branches)	lbann::model	private
get_weights()	lbann::model
get_weights() const	lbann::model
get_weights_pointers() const	lbann::model
grids	lbann::model	private
grids_mpi_groups	lbann::model	private
has_valid_execution_context() const noexcept	lbann::model	inline
insert_layer(OwningLayerPtr &&l, std::string const &parent_name)	lbann::model
is_subgraph_parallelism_enabled() const noexcept	lbann::model	inline
load_from_checkpoint_distributed(persist &p)	lbann::model
load_from_checkpoint_shared(persist &p)	lbann::model
m_background_io_allowed	lbann::model	private
m_callbacks	lbann::model	private
m_comm	lbann::model	private
m_current_mini_batch_size	lbann::model	private
m_default_optimizer_msg	lbann::model	private
m_execution_context	lbann::model	private
m_layers	lbann::model	private
m_max_mini_batch_size	lbann::model	private
m_metrics	lbann::model	private
m_model_is_setup	lbann::model	private
m_name	lbann::model	private
m_objective_function	lbann::model	private
m_weights	lbann::model	private
model(lbann_comm *comm, std::unique_ptr< objective_function > obj_fn, std::unique_ptr< lbann_data::Optimizer > default_optimizer_msg=nullptr)	lbann::model
model(const model &other)	lbann::model
model()	lbann::model	private
num_resources_branch_layers	lbann::model	private
num_resources_non_branch_layers	lbann::model	private
operator=(const model &other)	lbann::model
reconcile_weight_values()	lbann::model
remap_pointers(const std::unordered_map< Layer *, ViewingLayerPtr > &layer_map, const std::unordered_map< weights *, ViewingWeightsPtr > &weights_map)	lbann::model	private
remove_layer(std::string const &name)	lbann::model
remove_weights(std::string const &name)	lbann::model
reorder_layers(const std::vector< El::Int > &gather_indices)	lbann::model	private
replace_layer(OwningLayerPtr &&l, std::string const &name)	lbann::model
reset_epoch_statistics(execution_mode mode)	lbann::model
reset_mode(ExecutionContext &context, execution_mode mode)	lbann::model
save_model()	lbann::model
save_to_checkpoint_distributed(persist &p)	lbann::model
save_to_checkpoint_shared(persist &p)	lbann::model
serialize(Archive &ar)	lbann::model
set_current_mini_batch_size(El::Int) noexcept	lbann::model	inline
set_name(std::string name)	lbann::model
set_num_resources_branch_layers(int num) noexcept	lbann::model	inline
set_num_resources_non_branch_layers(int num) noexcept	lbann::model	inline
set_subgraph_num_parent_resources(int num_resources) noexcept	lbann::model	inline
set_subgrid_communication_type(int type) noexcept	lbann::model	inline
set_subgrid_topology(bool type) noexcept	lbann::model	inline
setup(size_t max_mini_batch_size, const std::vector< El::Grid *> &grids, bool force=false)	lbann::model
setup_layer_execution_order()	lbann::model	private
setup_layer_grid_tags(const std::vector< El::Grid *> &grids)	lbann::model	private
setup_layer_topology()	lbann::model	private
setup_layers(size_t max_mini_batch_size, const std::vector< El::Grid *> &grids)	lbann::model	private
setup_subcommunicators(const std::vector< El::Grid *> &grids)	lbann::model	private
setup_subgrid_layers_run_condition()	lbann::model	private
setup_subgrids()	lbann::model	private
setup_weights()	lbann::model	private
subCommunicatorsSubgrids	lbann::model	private
subgraph_num_resources_parent	lbann::model	private
summarize_matrices(lbann_summary &summarizer)	lbann::model
summarize_stats(lbann_summary &summarizer)	lbann::model
swap_layers(model &other)	lbann::model
swap_metrics(model &other)	lbann::model
swap_objective_function(model &other)	lbann::model
swap_weights(model &other)	lbann::model
update_layers()	lbann::model
update_weights()	lbann::model
vector_communication_subgraph	lbann::model	private
write_proto(lbann_data::Model &proto)	lbann::model
~model()=default	lbann::model