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

↓ 16 callersMethodget_secondary_grid_ranks

Return the ranks of secondary grid in the trainer */

include/lbann/comm.hpp:913

↓ 16 callersMethodget_type

src/models/unit_test/model_test.cpp:214

↓ 16 callersMethodsamples

include/lbann/utils/accumulating_timer.hpp:137

↓ 16 callersMethodset_cb_type

include/lbann/io/persist.hpp:92

↓ 16 callersMethodsubgraph_parallelism_execution

layer-level sub-graph parallelism execution (trb 10/03/2023): used

include/lbann/layers/layer.hpp:524

↓ 16 callersMethodswap

src/utils/cudnn.cpp:728

↓ 16 callersMethodweights

src/weights/weights.cpp:67

↓ 15 callersFunctionconv

(keras_layer, pb_layer)

model_zoo/models/python/keras/kerbann.py:125

↓ 15 callersMethodcopy

src/layers/learning/gru.cpp:87

↓ 15 callersMethodepsilon

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

↓ 15 callersMethodgather

include/lbann/comm_impl.hpp:359

↓ 15 callersMethodget_current_mini_batch_size

src/utils/dataset.cpp:99

↓ 15 callersMethodget_input_size

include/lbann/execution_algorithms/kfac/kfac_block.hpp:194

↓ 15 callersMethodget_local_thread_id

src/utils/threads/thread_pool.cpp:262

↓ 15 callersMethodinputs

Marks the given tensor as an input of the tested LBANN model, and returns a matching LBANN Input node (or a Slice/Reshape thereof).

ci_test/common_python/test_util.py:178

↓ 15 callersMethodmin

Binary math functions

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

↓ 15 callersMethodname

src/callbacks/set_weights_value.cpp:72

↓ 15 callersMethodoption_is_defined

include/lbann/utils/argument_parser.hpp:645

↓ 14 callersMethodGather

include/lbann/layers/transform/distconv/distconv_gather.hpp:46

↓ 14 callersFunctiondefault_hydrogen_stream

include/lbann/utils/distconv.hpp:58

↓ 14 callersMethodget_cb_type

include/lbann/io/persist.hpp:90

↓ 14 callersMethodget_mean_value

src/metrics/metric.cpp:71

↓ 14 callersMethodget_status

include/lbann/optimizers/optimizer.hpp:222

↓ 14 callersMethodget_sync_info

src/execution_algorithms/kfac/kfac_block.cpp:43

↓ 14 callersMethodglobal_barrier

src/comm.cpp:720

↓ 14 callersFunctionis_active

src/utils/python.cpp:96

↓ 14 callersMethodis_frozen

src/layers/layer.cpp:449

↓ 14 callersMethodload

src/data_readers/data_reader_csv.cpp:109

↓ 14 callersFunctionpow

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

↓ 14 callersFunctionread_filelist

src/utils/jag_common.cpp:33

↓ 14 callersMethodreduce_scalar

include/lbann/utils/summary_impl.hpp:129

↓ 14 callersMethodset_layer

src/utils/cudnn.cpp:1021

↓ 13 callersFunctionTranslateBetweenGridsVC

src/execution_algorithms/kfac/kfac_util.cpp:628

↓ 13 callersFunction_load_pretrained_weights

( *fn, file_dir=os.path.join( os.path.dirname(os.path.abspath(__file__)), "pretrained_weights"

python/lbann/models/roberta.py:36

↓ 13 callersMethodadd_weights

src/models/model.cpp:496

↓ 13 callersMethodcreate_optimizer

include/lbann/models/model.hpp:672

↓ 13 callersMethodfetch

src/data_readers/data_reader.cpp:86

↓ 13 callersMethodget_branch_tag_input

src/layers/data_type_layer.cpp:524

↓ 13 callersMethodget_data_schema

@brief this method is made public for testing */

include/lbann/data_readers/data_reader_HDF5.hpp:169

↓ 13 callersMethodget_metrics

src/models/model.cpp:332

↓ 13 callersMethodget_position

Get the current position in the data reader.

include/lbann/utils/dataset.hpp:146

↓ 13 callersFunctionget_used_gpu_memory

* @brief Returns the currently used memory, or 0 if LBANN was not compiled with * GPU support. */

src/callbacks/memory_profiler.cpp:49

↓ 13 callersMethodhelp_requested

include/lbann/utils/argument_parser.hpp:790

↓ 13 callersFunctionmake_default_handle_manager

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

↓ 13 callersFunctionmake_default_handle_manager

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

↓ 13 callersMethodprint

src/utils/timer_map.cpp:94

↓ 13 callersMethodpush

@brief Adds a value to back of the queue */

include/lbann/utils/threads/thread_safe_queue.hpp:70

↓ 13 callersMethodresize

Resize to h x w. */

include/lbann/transforms/vision/resize.hpp:42

↓ 13 callersMethodroot

include/lbann/utils/serialization/rooted_archive_adaptor.hpp:151

↓ 13 callersMethodset_comm

src/weights/weights.cpp:197

↓ 13 callersFunctionsystem

Name of system. Hostname with trailing digits removed.

python/lbann/contrib/lc/systems.py:36

↓ 13 callersMethodusing_gpus

include/lbann/layers/layer.hpp:417

↓ 13 callersMethodwrite

Write script to file. The working directory is created if needed. Args: overwrite (bool): Whether to overwrite script fi

python/lbann/launcher/batch_script.py:108

↓ 13 callersMethodwrite_proto

src/optimizers/sgd.cpp:114

↓ 12 callersFunctionTranslateBetweenGridsVCAsyncDirect

src/execution_algorithms/kfac/kfac_util.cpp:998

↓ 12 callersMethodall_gather_packed_lists

include/lbann/data_readers/sample_list_impl.hpp:945

↓ 12 callersFunctionbuild_model_from_prototext

src/utils/lbann_library.cpp:401

↓ 12 callersFunctionconstruct_all_options

src/utils/options.cpp:569

↓ 12 callersMethodcopy_trained_weights_from

src/models/model.cpp:569

↓ 12 callersFunctioncreate_parallel_strategy

(num_channel_groups)

ci_test/unit_tests/test_unit_layer_batched_matmul_distconv.py:38

↓ 12 callersFunctiondesc_string

src/optimizers/unit_test/optimizer_common.hpp:44

↓ 12 callersMethoddisable_labels

Disable fetching labels.

include/lbann/data_readers/data_reader_csv.hpp:64

↓ 12 callersMethodget

@returns @a Borrowed reference. */

include/lbann/utils/python.hpp:143

↓ 12 callersMethodget_communication_flag

(trb 10/03/2023): used, keeping setter, which is unused.

include/lbann/layers/layer.hpp:489

↓ 12 callersMethodget_current_batch_size

include/lbann/execution_algorithms/kfac/kfac_block.hpp:192

↓ 12 callersMethodget_file_dir

src/data_readers/data_reader.cpp:731

↓ 12 callersMethodget_num_data

src/data_readers/data_reader_jag_conduit.cpp:1247

↓ 12 callersFunctionget_packed_strides

include/lbann/utils/dim_helpers.hpp:91

↓ 12 callersMethodinitialized

include/lbann/utils/dataset.hpp:168

↓ 12 callersMethodis_root

tools/compute_mean/mpi_states.hpp:67

↓ 12 callersMethodname

include/lbann/callbacks/hang.hpp:58

↓ 12 callersMethodscale

Scale all data by scale_val. */

include/lbann/transforms/scale.hpp:44

↓ 12 callersMethodset_data_filename

src/data_readers/data_reader.cpp:757

↓ 12 callersMethodstart

@brief Start counting time. * * If the clock is already running, this will restart the counter. * * @post running() returns @c true; sto

include/lbann/utils/timer.hpp:80

↓ 12 callersFunctionsystem

Name of system. Hostname with trailing digits removed.

python/lbann/contrib/olcf/systems.py:30

↓ 11 callersMethodabort

Abort MPI. Only a designated rank prints out a message.

tools/compute_mean/mpi_states.cpp:48

↓ 11 callersMethodadd_parallel_command

Add command to be executed in parallel. The command is launched with mpiexec. Parallel processes are distributed evenly amongst the c

python/lbann/launcher/pjm.py:82

↓ 11 callersMethodadd_transform

* Add trans as the next transform to apply. */

include/lbann/transforms/transform_pipeline.hpp:55

↓ 11 callersFunctionfast_rand_int

include/lbann/utils/random.hpp:56

↓ 11 callersMethodfetch_active_batch_synchronous

src/data_coordinator/buffered_data_coordinator.cpp:266

↓ 11 callersMethodget_experiment_schema

@brief this method is made public for testing */

include/lbann/data_readers/data_reader_HDF5.hpp:167

↓ 11 callersFunctionglob

* Wrapper around glob, which searches for paths matching pattern according to * the shell. * Note this does not do tilde expansion. */

include/lbann/utils/glob.hpp:42

↓ 11 callersFunctioninfinity

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

↓ 11 callersFunctioninput

(keras_layer, pb_layer)

model_zoo/models/python/keras/kerbann.py:114

↓ 11 callersMethodoutput_dims

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

↓ 11 callersMethodset_conduit_node

n.b. Do not put any PROFILE or DEBUG_DS statements in this method, since the threading from the data_reader will cause you grief

src/data_store/data_store_conduit.cpp:386

↓ 11 callersMethodsetup

src/data_readers/data_reader.cpp:78

↓ 11 callersMethodsetup

src/layers/layer.cpp:464

↓ 11 callersFunctionsystem

Name of system. Hostname with trailing digits removed.

python/lbann/contrib/nersc/systems.py:29

↓ 11 callersMethodto_string

src/data_readers/data_reader_jag_conduit.cpp:1275

↓ 10 callersMethodScatter

include/lbann/layers/transform/distconv/distconv_scatter.hpp:46

↓ 10 callersMethodadd_child_layer

src/layers/layer.cpp:952

↓ 10 callersFunctionadd_delimiter

* This automatically attaches the directory deliminator at the end of the given * directory as necessary. * If "" is given, it will do nothing */

src/utils/file_utils.cpp:122

↓ 10 callersMethodadd_to_gradient

include/lbann/optimizers/optimizer_impl.hpp:200

↓ 10 callersFunctionconstruct_trainer

src/proto/factories/trainer_factory.cpp:40

↓ 10 callersFunctiondirectory_exists

src/utils/file_utils.cpp:240

↓ 10 callersFunctiondirname

ci_test/integration_tests/conftest.py:27

↓ 10 callersMethodget

src/utils/unit_test/factory_test.cpp:78

↓ 10 callersMethodget_index_owner

src/data_store/data_store_conduit.cpp:900

↓ 10 callersMethodget_intertrainer_master

Return the rank of the inter-trainer master process. */

include/lbann/comm.hpp:188

← previous next →201–300 of 8,063, ranked by callers