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

↓ 10 callersMethodget_kernel_dims

src/layers/learning/convolution.cpp:118

↓ 10 callersMethodget_num_iterations_per_epoch

src/data_coordinator/data_coordinator.cpp:514

↓ 10 callersMethodget_output_size

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

↓ 10 callersMethodget_type

include/lbann/layers/transform/scatter.hpp:182

↓ 10 callersMethodinput_dims

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

↓ 10 callersMethodis_subgraph_parallelism_enabled

include/lbann/models/model.hpp:714

↓ 10 callersFunctionisfinite

src/optimizers/adam.cpp:39

↓ 10 callersFunctionload_file

Load a file into a buffer

src/utils/file_utils.cpp:188

↓ 10 callersMethodparse_schemas

src/data_readers/unit_test/data_reader_common_HDF5_test_utils.hpp:58

↓ 10 callersMethodrecv

src/comm.cpp:762

↓ 10 callersMethodsend

src/comm.cpp:728

↓ 10 callersFunctionset_environment

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

↓ 10 callersMethodsetup

src/models/model.cpp:706

↓ 10 callersFunctionsplit

src/utils/profiling.cpp:85

↓ 10 callersMethodstep

include/lbann/optimizers/data_type_optimizer_impl.hpp:199

↓ 10 callersMethodstop

@brief Get the total elapsed time in seconds. * * Resets the counter, so a subsequent call to stop() or check() * without another start() w

include/lbann/utils/timer.hpp:89

↓ 9 callersFunctionEntrywiseZipInto

src/operators/math/common.hpp:38

↓ 9 callersMethod__init__

(self, config, add_pooling_layer=True, load_weights=True)

python/lbann/models/roberta.py:544

↓ 9 callersMethodapply

src/transforms/scale.cpp:34

↓ 9 callersMethodencoder

applications/physics/ICF/macc_network_architectures.py:125

↓ 9 callersMethodforward_prop

src/models/model.cpp:1553

↓ 9 callersMethodget_name

include/lbann/models/model.hpp:621

↓ 9 callersFunctionget_neighbors

src/utils/graph.cpp:50

↓ 9 callersMethodinit

include/lbann/utils/argument_parser.hpp:741

↓ 9 callersMethodkeep_sample_order

src/data_readers/data_reader.cpp:748

↓ 9 callersFunctionload

@brief Load this half-precision value from Binary */

src/utils/serialization.cpp:52

↓ 9 callersMethodready_for_next_fetch

src/data_coordinator/buffered_data_coordinator.cpp:374

↓ 9 callersMethodreset

src/utils/miopen.cpp:179

↓ 9 callersMethodset_dims

Set weight tensor dimensions as a 1D tensor. */

include/lbann/weights/weights.hpp:175

↓ 9 callersMethodset_label_filename

src/data_readers/data_reader.cpp:769

↓ 9 callersMethodset_num_weights

include/lbann/layers/layer.hpp:741

↓ 9 callersMethodset_role

src/data_readers/data_reader.cpp:929

↓ 9 callersMethodsetup

Setup metric. */

include/lbann/metrics/metric.hpp:115

↓ 9 callersFunctionshuffle

model_zoo/jag_utils/python/shuffle.py:5

↓ 9 callersFunctionsplice_dims

include/lbann/utils/dim_helpers.hpp:142

↓ 9 callersFunctionto_miopen

@brief Convert a LBANN forward convolution algorithm to the MIOpen * equivalent value. */

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

↓ 9 callersMethodtrain

src/trainers/trainer.cpp:242

↓ 9 callersMethodweights_names

@name Data access and query */ @{

src/callbacks/ltfb.cpp:113

↓ 8 callersFunctionAsFFTWType

include/lbann/utils/fftw_wrapper.hpp:68

↓ 8 callersMethodKFAC

@todo Initialize properly

src/execution_algorithms/kfac.cpp:57

↓ 8 callersMethod__init__

Initialize ResNet. Args: block (type): Residual block type, which should be a `lbann.modules.Module`.

python/lbann/models/resnet.py:231

↓ 8 callersMethod__init__

Initialize ResNet. Args: block (type): Residual block type, which should be a `lbann.modules.Module`.

applications/selfsupervised/resnet.py:233

↓ 8 callersMethodadd_parent_layer

src/layers/layer.cpp:938

↓ 8 callersMethodbackward

src/layers/math/distconv/distconv_matmul.cpp:125

↓ 8 callersFunctionbatch_step_string

Get human-readable string describing current batch step. */

src/callbacks/debug.cpp:80

↓ 8 callersMethodcheck

@brief Get the current elapsed time (seconds) without stopping. */

include/lbann/utils/timer.hpp:97

↓ 8 callersMethodcheck_mpi

Check the return code of an MPI call, and abort if not successful.

tools/compute_mean/mpi_states.cpp:38

↓ 8 callersMethodcompare

Comparison operation to sort vector entries. * Entries are sorted by value in decreasing order, with ties * broken in favor of entries with sm

src/layers/transform/in_top_k.cpp:59

↓ 8 callersFunctioncopy_all

src/layers/data_type_layer.cpp:46

↓ 8 callersFunctiondump_network

Dump the local network matrices for debugging. * Dump only the local matrices because not every rank will * necessarily have bad data, and the che

src/callbacks/check_nan.cpp:184

↓ 8 callersMethodencode

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

↓ 8 callersMethodexport_proto

Get a protobuf representation of this object.

python/lbann/core/training_algorithm.py:201

↓ 8 callersFunctionfini_comm

Destroy a communicator. */

model_zoo/tests/comm_test.cpp:48

↓ 8 callersMethodfreeze

src/layers/layer.cpp:433

↓ 8 callersMethodget_branch_tag_input_vector

src/layers/data_type_layer.cpp:542

↓ 8 callersMethodget_data_store_ptr

include/lbann/data_readers/data_reader.hpp:451

↓ 8 callersMethodget_device_allocation

src/layers/learning/gru.cpp:110

↓ 8 callersMethodget_effective_num_ranks

Return the num of ranks that only have assigned data.

tools/compute_mean/image_list.hpp:72

↓ 8 callersMethodget_hint_layer

src/layers/layer.cpp:431

↓ 8 callersFunctionget_matrix_inverse

src/execution_algorithms/kfac/kfac_util.cpp:198

↓ 8 callersMethodget_num_data

src/data_coordinator/unit_test/buffered_data_coordinator_test.cpp:61

↓ 8 callersMethodget_num_samples

src/data_coordinator/data_coordinator.cpp:451

↓ 8 callersFunctionget_num_spatial_dims

src/utils/distconv.cpp:505

↓ 8 callersFunctionget_rank_in_world

src/comm.cpp:929

↓ 8 callersMethodget_raw_sample

src/data_readers/data_reader_smiles.cpp:561

↓ 8 callersMethodhas_bias

include/lbann/layers/learning/channelwise_fully_connected.hpp:135

↓ 8 callersFunctioninit_comm

Initialize a new communicator. */

model_zoo/tests/comm_test.cpp:43

↓ 8 callersMethodis_execution_mode_valid

src/data_coordinator/data_coordinator.cpp:481

↓ 8 callersFunctionmake_test

(name, test_by_platform_list=[], args=[])

ci_test/unit_tests/test_unit_datastore_imagenet.py:264

↓ 8 callersFunctionpack_lower_tri

src/execution_algorithms/kfac/kfac_util.cpp:1525

↓ 8 callersFunctionrank_string

Get human-readable string describing process rank. */

src/callbacks/debug.cpp:45

↓ 8 callersMethodreset

src/metrics/metric.cpp:57

↓ 8 callersMethodrunning

@brief Check if the timer is running. */

include/lbann/utils/timer.hpp:109

↓ 8 callersMethodruns_inplace

@brief If true, the layer will run in-place (the input * and output activations point to the same tensor). * Value is set during graph setup (in

include/lbann/layers/layer.hpp:704

↓ 8 callersMethodsample_list

include/lbann/data_readers/sample_list_impl.hpp:293

↓ 8 callersMethodsendrecv

include/lbann/comm_impl.hpp:923

↓ 8 callersMethodset_data_reader_ptr

src/data_store/data_store_conduit.cpp:190

↓ 8 callersMethodset_mini_batch_size

src/utils/dataset.cpp:125

↓ 8 callersMethodset_offset

src/data_readers/data_reader_smiles.cpp:843

↓ 8 callersMethodset_preloaded_conduit_node

src/data_store/data_store_conduit.cpp:305

↓ 8 callersFunctionsetup_layer

src/layers/unit_test/example_layer.cpp:158

↓ 8 callersMethodstddev

include/lbann/utils/accumulating_timer.hpp:147

↓ 8 callersMethodsync

include/lbann/layers/transform/distconv/distconv_nvshmem_vector_addressing.hpp:187

↓ 8 callersFunctionto_cudnn

@brief Convert a LBANN forward convolution algorithm to the cuDNN * equivalent value. */

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

↓ 8 callersFunctionunpack_lower_tri

src/execution_algorithms/kfac/kfac_util.cpp:1539

↓ 8 callersFunctionupdate_kronecker_average

src/execution_algorithms/kfac/kfac_util.cpp:1503

↓ 8 callersFunctionzeros

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

↓ 7 callersMethodEnd

tools/compute_mean/Mat.hpp:36

↓ 7 callersMethodGet

tools/compute_mean/Mat.hpp:173

↓ 7 callersMethodLTFB

@name Life-cycle management */ @{ @brief Construct LTFB from its component pieces. * @param[in] name A string identifying this instance of LTFB.

include/lbann/execution_algorithms/ltfb.hpp:83

↓ 7 callersMethod__init__

applications/nas/nasnet/search/operations.py:153

↓ 7 callersMethod_get_subgraph

Returns a parallel strategy based on the attention head subgraph parallelism configuration and the requested grid tag. :para

python/lbann/modules/transformer.py:253

↓ 7 callersMethoda

include/lbann/utils/beta.hpp:92

↓ 7 callersFunctionabs

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

↓ 7 callersFunctionanderson_darling_test

@brief Statistic for Anderson-Darling normality test * * A higher statistic implies greater confidence in non-normality. * * See: * * George

src/utils/unit_test/random_fill_test.cpp:45

↓ 7 callersMethodb

include/lbann/utils/beta.hpp:93

↓ 7 callersFunctionbroadcast_variable_grids

Broadcast the variabes in sub-grid parallelism

src/execution_algorithms/kfac.cpp:1185

↓ 7 callersMethodcompute_forward

include/lbann/utils/fftw_wrapper.hpp:313

↓ 7 callersMethodconstant

(self, value, dims=[], datatype=None, hint_layer=None)

applications/ATOM/models/wae.py:407

↓ 7 callersMethodconstant

(self, value, dims=[], datatype=None, hint_layer=None)

applications/ATOM/models/vae.py:370

← previous next →301–400 of 8,063, ranked by callers