Functions — github.com/LBANN/lbann

CodeHub MCPcopy Create free account

hub / github.com/LBANN/lbann / functions

Functions8,063 in github.com/LBANN/lbann

⨍ Functions 8,063◇ Types & classes 1,225↳ Endpoints 4

↓ 37 callersMethodget

include/lbann/utils/dnn_lib/cudnn/utils.hpp:70

↓ 37 callersMethodget_values_sharded

src/weights/data_type_weights.cpp:320

↓ 37 callersFunctionjoin_path

include/lbann/utils/file_utils.hpp:147

↓ 37 callersMethodmark_updated

src/layers/distconv_adapter.cpp:463

↓ 37 callersMethodsize

include/lbann/data_readers/sample_list_open_files_impl.hpp:113

↓ 36 callersMethodget_values

src/weights/data_type_weights.cpp:303

↓ 36 callersMethodmark_invariant

src/layers/distconv_adapter.cpp:468

↓ 35 callersMethodget_child_layers

src/layers/layer.cpp:891

↓ 35 callersMethodget_data_layout

src/layers/learning/gru.cpp:104

↓ 35 callersMethodget_gradient_buffer

src/execution_algorithms/kfac/kfac_block_gru.cpp:1102

↓ 35 callersMethodget_procs_in_world

Return the total number of ranks. */

include/lbann/comm.hpp:229

↓ 35 callersFunctionhash_combine

include/lbann/utils/hash.hpp:47

↓ 35 callersFunctionload_parameter

Create a PyTorch Parameter object with weights from LBANN. Weight file is assumed to have been created by the "dump weights" callback in LBAN

applications/nlp/transformer/evaluate_translation_bleu.py:95

↓ 34 callersMethodexists

@todo Deprecated.

src/io/file_io.cpp:57

↓ 34 callersMethodfp_compute

src/layers/misc/argmin.cpp:68

↓ 34 callersMethodget_batch_interval

@brief Return the batch interval. */

include/lbann/callbacks/callback.hpp:200

↓ 34 callersMethodsize

include/lbann/data_readers/sample_list_impl.hpp:409

↓ 34 callersMethodtrainer_barrier

src/comm.cpp:714

↓ 33 callersFunctionbuild_string

include/lbann/utils/exception.hpp:160

↓ 33 callersFunctioncheck_error

src/utils/python.cpp:98

↓ 32 callersMethoddistconv_enabled

src/layers/layer.cpp:1096

↓ 32 callersMethodget_trainer_master

Return the rank of the master process in this trainer. */

include/lbann/comm.hpp:186

↓ 32 callersMethodintertrainer_gather

include/lbann/comm_impl.hpp:331

↓ 32 callersMethodset_weights

include/lbann/layers/layer.hpp:742

↓ 31 callersMethodadd_header_line

Add line to script header. The header should specify configuration options for the job scheduler, without containing executable comma

python/lbann/launcher/batch_script.py:52

↓ 31 callersFunctionswap

src/utils/cudnn.cpp:180

↓ 30 callersMethodbp_compute

src/layers/math/matmul.cpp:97

↓ 30 callersMethodreset_mode

At the start of the epoch, set the execution mode and make sure that each layer points to this model

src/models/model.cpp:1513

↓ 29 callersFunctiontest_func

Function that can interact with PyTest. Returns a dict containing log files and other output data.

ci_test/common_python/tools.py:772

↓ 29 callersMethodwait

src/comm.cpp:636

↓ 28 callersMethodget_input_size

src/layers/layer.cpp:314

↓ 28 callersMethodget_num_trainers

Return the total number of trainers. */

include/lbann/comm.hpp:220

↓ 27 callersMethodGet

src/layers/regularizers/local_response_normalization.cpp:45

↓ 27 callersMethodMakeEmpty

src/layers/matrix_builder.hpp:90

↓ 27 callersMethodget_grid_tag

src/layers/layer.cpp:254

↓ 27 callersMethodget_layer

Get the layer being managed. */

include/lbann/utils/dnn_lib/dnn_lib.hpp:580

↓ 27 callersMethodget_matrix_height

src/weights/weights.cpp:154

↓ 26 callersMethodget_matrix_width

src/weights/weights.cpp:159

↓ 26 callersMethodget_num_threads

@brief Query the number of worker threads actually present */

include/lbann/utils/threads/thread_pool.hpp:145

↓ 25 callersMethodget_gradient_sharded

include/lbann/optimizers/data_type_optimizer_impl.hpp:120

↓ 25 callersMethodget_output_size

src/layers/layer.cpp:340

↓ 25 callersMethodget_role

* Get the role for this dataset. */

include/lbann/utils/dataset.hpp:71

↓ 25 callersMethodset_dims

src/layers/transform/permute/cutt_permuteimpl.hpp:251

↓ 25 callersMethodtrainer_broadcast

include/lbann/comm_impl.hpp:48

↓ 24 callersMethoddecode

applications/graph/node2vec/model/random_projection.py:137

↓ 24 callersFunctionget_sync_info

include/lbann/utils/sync_info_helpers.hpp:39

↓ 24 callersMethodsave

src/layers/transform/cereal_registration/permute.cpp:42

↓ 24 callersFunctionto_string

include/lbann/io/persist_impl.hpp:59

↓ 24 callersMethodtrainer_reduce

include/lbann/comm_impl.hpp:470

↓ 23 callersMethodallreduce

src/comm.cpp:559

↓ 23 callersMethodhas_weights

include/lbann/layers/layer.hpp:736

↓ 23 callersMethodset_optimizer

src/weights/data_type_weights.cpp:216

↓ 23 callersMethodset_value

src/weights/data_type_weights.cpp:359

↓ 22 callersMethodevaluate

NOTE (trb 04/19/2021): Currently, "evaluate" is just defined as "run forward prop and look at objective functions/metrics/etc". This is currently impl

src/trainers/trainer.cpp:285

↓ 22 callersMethodfind_child_layer_index

src/layers/layer.cpp:919

↓ 22 callersFunctioninit_random

src/utils/random_number_generators.cpp:164

↓ 22 callersMethodset_initializer

src/weights/data_type_weights.cpp:175

↓ 21 callersFunctionexp

python/lbann/torch/replacements/functions.py:282

↓ 21 callersMethodget_comm

include/lbann/models/model.hpp:650

↓ 21 callersMethodget_max

include/lbann/operators/math/clamp.hpp:106

↓ 21 callersMethodget_min

include/lbann/operators/math/clamp.hpp:105

↓ 21 callersMethodget_objective_function

include/lbann/models/model.hpp:623

↓ 21 callersFunctioninit_data_seq_random

src/utils/random_number_generators.cpp:204

↓ 21 callersMethodmodel

src/models/model.cpp:72

↓ 20 callersMethodadd_command

Add executable command to script. Args: command (`str` or `Iterable` of `str`s): Program invocation or sequence o

python/lbann/launcher/batch_script.py:69

↓ 20 callersFunctionconstruct_std_options

src/utils/options.cpp:33

↓ 20 callersMethodget_callbacks

include/lbann/models/model.hpp:634

↓ 20 callersFunctionget_mpi_comm

src/utils/distconv.cpp:331

↓ 20 callersMethodget_weights

include/lbann/optimizers/data_type_optimizer_impl.hpp:75

↓ 20 callersMethodmake_reference

Marks the given tensor as a reference output of the tested LBANN model, and returns a matching LBANN node. :param ref: The r

ci_test/common_python/test_util.py:209

↓ 20 callersFunctionset_environment

python/lbann/contrib/lc/launcher.py:45

↓ 20 callersMethodset_matrix_distribution

src/weights/weights.cpp:192

↓ 19 callersFunctionRowMajor

src/layers/transform/permute/tensor_dims_utils.hpp:158

↓ 19 callersMethodSet

tools/compute_mean/Mat.hpp:167

↓ 19 callersFunctioncommify

src/utils/commify.cpp:34

↓ 19 callersMethoddistconv_enabled

@brief Indicate whether distconv is enabled. */

include/lbann/layers/layer.hpp:1090

↓ 19 callersMethodget_max_mini_batch_size

include/lbann/models/model.hpp:739

↓ 19 callersFunctionget_number_of_io_partitions

src/utils/distconv.cpp:380

↓ 19 callersFunctionones

src/transforms/vision/unit_test/helper.hpp:75

↓ 19 callersMethodset_loss

Sets an LBANN node to be measured for the test. :param loss: The LBANN graph node to use for the test. :param tolerance: Opt

ci_test/common_python/test_util.py:268

↓ 19 callersFunctionsetup

(data_reader_file, name='classifier', num_labels=200, mini_batch_size=128,

applications/selfsupervised/classifier.py:7

↓ 18 callersMethodadd_weights

@name Weights-related accessors */ @{

include/lbann/layers/layer.hpp:731

↓ 18 callersFunctiondesc

src/layers/transform/unit_test/permute_layer_test.cpp:57

↓ 18 callersFunctionguess_global_rank

model_zoo/lbann.cpp:52

↓ 18 callersFunctionisnan

include/lbann/utils/impl/gpu_lib.hpp:105

↓ 18 callersMethodmax

include/lbann/utils/impl/rocm.hpp:229

↓ 18 callersFunctionset_io_generators_local_index

src/utils/random_number_generators.cpp:116

↓ 18 callersFunctionsystem

Name of compute system.

ci_test/common_python/tools.py:1031

↓ 17 callersFunctionconstruct_jag_options

src/utils/options.cpp:489

↓ 17 callersFunctiondirname

applications/MOF/test/conftest.py:30

↓ 17 callersMethodget_model

@brief Get a reference to the model that manages this layer. * * May be null if this layer is "free" (e.g., during model assembly * or for

include/lbann/layers/layer.hpp:341

↓ 17 callersFunctionget_num_dims

src/utils/distconv.cpp:493

↓ 17 callersMethodmark_equivalent

src/layers/distconv_adapter.cpp:449

↓ 17 callersMethodraw_value

include/lbann/utils/environment_variable.hpp:120

↓ 16 callersMethodallocate

include/lbann/utils/impl/rocm.hpp:270

↓ 16 callersFunctionepsilon

include/lbann/utils/impl/cuda.hpp:248

↓ 16 callersFunctionfinalize

src/utils/python.cpp:72

↓ 16 callersMethodget

include/lbann/utils/dnn_lib/miopen/utils.hpp:70

↓ 16 callersMethodget_primary_grid_ranks

Return the ranks of primary grid in the trainer */

include/lbann/comm.hpp:910

↓ 16 callersMethodget_sample_stride

Return the sample stride.

include/lbann/utils/dataset.hpp:118

← previous next →101–200 of 8,063, ranked by callers