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

↓ 7 callersFunctionconv

(x, dtype, **kwargs)

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

↓ 7 callersMethodcreate

src/utils/cudnn.cpp:207

↓ 7 callersMethoddecode_smiles

src/data_readers/data_reader_smiles.cpp:489

↓ 7 callersMethoddecoder

applications/physics/ICF/macc_network_architectures.py:141

↓ 7 callersFunctiondirname

ci_test/unit_tests/conftest.py:31

↓ 7 callersMethodfind_parent_layer_index

src/layers/layer.cpp:900

↓ 7 callersFunctionfunc

python/lbann/torch/converters.py:89

↓ 7 callersMethodget_backprop_requirements

* @brief Returns the necessary tensors for computing backpropagation */

include/lbann/layers/layer.hpp:391

↓ 7 callersFunctionget_file_names

(dir_name, test_name)

ci_test/unit_tests/test_unit_lbann_invocation.py:19

↓ 7 callersFunctionget_input_dims

src/utils/unit_test/fftw_test.cpp:42

↓ 7 callersMethodget_inverse_proc_rank

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

↓ 7 callersFunctionget_matrix_inverse_eigen

src/execution_algorithms/kfac/kfac_util.cpp:265

↓ 7 callersFunctionget_r2c_output_dims

include/lbann/utils/fft_common.hpp:69

↓ 7 callersFunctionget_time

Return time in fractional seconds since an epoch. */

tools/compute_mean/walltimes.hpp:38

↓ 7 callersMethodget_type

include/lbann/layers/transform/gather.hpp:178

↓ 7 callersMethodget_type

include/lbann/layers/transform/slice.hpp:181

↓ 7 callersMethodget_unit

Return a display unit for this metric. * Default is an empty string. This is overriden if the metric has * units, e.g. "%" or "sec". */

include/lbann/metrics/metric.hpp:112

↓ 7 callersFunctionhas_layer

src/models/unit_test/modify_test.cpp:76

↓ 7 callersFunctioninvert

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

↓ 7 callersMethodis_background_fetching_in_progress

include/lbann/data_coordinator/io_data_buffer.hpp:105

↓ 7 callersFunctionkeras_tuple_to_protobuf

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

↓ 7 callersFunctionmake_directory

src/utils/file_utils.cpp:249

↓ 7 callersFunctionmake_random_array

Hacked function to generate a random array. NumPy's RNG produces different values with different NumPy versions. This function is helpful whe

ci_test/unit_tests/test_unit_algo_kfac.py:22

↓ 7 callersMethodopen_restart

src/io/persist.cpp:309

↓ 7 callersFunctionsave

@brief Save this half-precision value in Binary */

src/utils/serialization.cpp:46

↓ 7 callersMethodset_background_fetching_in_progress

include/lbann/data_coordinator/io_data_buffer.hpp:100

↓ 7 callersMethodset_current_mini_batch_size

include/lbann/models/model.hpp:749

↓ 7 callersMethodset_has_labels

Whether or not a data reader has labels

include/lbann/data_readers/data_reader.hpp:335

↓ 7 callersMethodset_has_responses

Whether or not a data reader has a response field

include/lbann/data_readers/data_reader.hpp:340

↓ 7 callersMethodset_initial_position

Set the current position based on the base and model offsets

include/lbann/utils/dataset.hpp:140

↓ 7 callersMethodset_scale

Set scaling factor. */

include/lbann/layers/transform/evaluation.hpp:50

↓ 7 callersMethodsetup

src/weights/weights.cpp:219

↓ 7 callersFunctionsystem

Name of system. Hostname with trailing digits removed.

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

↓ 7 callersMethodtotal

include/lbann/utils/running_statistics.hpp:129

↓ 7 callersMethodvariance

include/lbann/utils/running_statistics.hpp:134

↓ 6 callersFunctionAbs

src/layers/misc/dft_abs.cpp:106

↓ 6 callersMethodBeg

tools/compute_mean/Mat.hpp:35

↓ 6 callersMethodIsInitialized

tools/compute_mean/Mat.hpp:25

↓ 6 callersFunctionTranslateBetweenGridsVCAsync

src/execution_algorithms/kfac/kfac_util.cpp:875

↓ 6 callersFunction__swapEndianInt

include/lbann/utils/file_utils.hpp:85

↓ 6 callersMethodadd_value

src/metrics/metric.cpp:40

↓ 6 callersMethodallreduce

include/lbann/comm_impl.hpp:643

↓ 6 callersMethodam_root

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

↓ 6 callersFunctioncheck_node_fields

src/data_readers/unit_test/data_reader_common_HDF5_test_utils.cpp:34

↓ 6 callersMethodconstant

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

ci_test/common_python/models/atom/wae.py:366

↓ 6 callersMethodcopy

src/models/unit_test/model_test.cpp:209

↓ 6 callersMethoddistribute_from_local_matrix

src/data_coordinator/buffered_data_coordinator.cpp:540

↓ 6 callersMethodencode_smiles

src/data_readers/data_reader_smiles.cpp:427

↓ 6 callersMethodexport_proto

Construct and return a protobuf message.

python/lbann/core/layer.py:59

↓ 6 callersFunctionget_default_parameters

(dir_name, two_models=True)

ci_test/unit_tests/test_unit_lbann_invocation.py:7

↓ 6 callersMethodget_matrix_distribution

src/weights/weights.cpp:191

↓ 6 callersMethodget_my_rank

tools/compute_mean/mpi_states.hpp:84

↓ 6 callersMethodget_num_files

include/lbann/data_readers/sample_list_open_files_impl.hpp:119

↓ 6 callersMethodget_num_layers

src/models/model.cpp:350

↓ 6 callersMethodget_num_spliting_groups

(trb 10/03/2023): used

include/lbann/layers/layer.hpp:501

↓ 6 callersFunctionget_parallel_strategy_args

A wrapper function to create parallel_strategy arguments for Distconv-enabled layers. Args: {sample, depth, height, width, channel, f

python/lbann/core/util.py:287

↓ 6 callersMethodget_procs_per_node

Return the number of processes in a compute node. */

include/lbann/comm.hpp:227

↓ 6 callersMethodget_rank

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

↓ 6 callersMethodget_size

src/weights/weights.cpp:145

↓ 6 callersMethodget_type

src/layers/misc/uniform_hash.cpp:46

↓ 6 callersMethodget_type

src/layers/learning/channelwise_fully_connected.cpp:400

↓ 6 callersFunctionis_cosmoflow_parallel_io_enabled

src/utils/distconv.cpp:382

↓ 6 callersFunctionis_execution_mode_hook

src/utils/visitor_hooks.cpp:32

↓ 6 callersMethodlaunch_pinned_threads

FIXME (trb 08/03/2019): Setting thread affinity is not a portable pthread operation (hence the _np suffix); indeed, OSX does not support it. Unfortuna

src/utils/threads/thread_pool.cpp:109

↓ 6 callersMethodload_on_root

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

↓ 6 callersFunctionload_prototext

model_zoo/models/python/expandMotifs.py:51

↓ 6 callersMethodmetric

src/metrics/metric.cpp:63

↓ 6 callersFunctionmod

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

↓ 6 callersFunctionparse_path

@todo Deprecated.

src/utils/file_utils.cpp:80

↓ 6 callersFunctionprint_leading_spaces

data_utils/hdf5/generate_schema_and_sample_list.cpp:81

↓ 6 callersMethodprint_report

src/utils/exception.cpp:62

↓ 6 callersMethodrelease

src/utils/cudnn.cpp:193

↓ 6 callersMethodreset_statistics

Clear all statistics. */

include/lbann/metrics/metric.hpp:126

↓ 6 callersFunctionselect_things_by_name

src/callbacks/helpers.hpp:35

↓ 6 callersMethodset_check_gradients_tensor

Sets the tensor to be used as the objective function when running the check gradients callback. When provided a non-scalar tensor, th

ci_test/common_python/test_util.py:240

↓ 6 callersFunctionset_environment

python/lbann/contrib/riken/launcher.py:57

↓ 6 callersMethodset_keep_prob

@brief set prob for keep each unit. */

include/lbann/layers/regularizers/dropout.hpp:129

↓ 6 callersMethodset_last_mini_batch_size

Set the last mini batch size

include/lbann/utils/dataset.hpp:124

↓ 6 callersMethodset_learning_rate

src/callbacks/learning_rate.cpp:154

↓ 6 callersMethodset_profile_msg

src/data_store/data_store_conduit.cpp:2320

↓ 6 callersMethodset_sample_list_name

include/lbann/data_readers/sample_list_impl.hpp:278

↓ 6 callersMethodset_status

include/lbann/optimizers/optimizer.hpp:223

↓ 6 callersMethodsetup_backward

@brief Setup the backward (inverse) transform. * @param in Input array; must be allocated, could be overwritten. * @param out Output array; mu

include/lbann/utils/fftw_wrapper.hpp:290

↓ 6 callersMethodsetup_forward

@brief Setup the forward transform. * @param in Input array; must be allocated, could be overwritten. * @param out Output array; must be alloc

include/lbann/utils/fftw_wrapper.hpp:260

↓ 6 callersFunctionswap

src/utils/miopen.cpp:174

↓ 6 callersMethodswap

src/utils/miopen.cpp:545

↓ 6 callersMethodtrainer

src/trainers/trainer.cpp:59

↓ 6 callersMethodtrainer_allreduce

include/lbann/comm_impl.hpp:628

↓ 6 callersMethodunfreeze

src/layers/layer.cpp:441

↓ 6 callersFunctionvalidate_mat

Validate every local entry of a matrix has a given value. */

model_zoo/tests/comm_test.cpp:59

↓ 6 callersMethodwait

include/lbann/comm_impl.hpp:754

↓ 6 callersFunctionwrite_column

src/data_coordinator/data_packer.cpp:57

↓ 6 callersFunctionwrite_file

src/data_readers/unit_test/data_reader_common_catch2.cpp:89

↓ 5 callersFunctionadd

(keras_layer, pb_layer)

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

↓ 5 callersMethodadd_callback

src/models/model.cpp:547

↓ 5 callersFunctionallocate_trainer_resources

Split the MPI communicator into trainers Return the

src/utils/lbann_library.cpp:74

↓ 5 callersMethodbackground_io_activity_allowed

@brief Are background I/O activities enabled by the data coordinator */

include/lbann/trainers/trainer.hpp:201

↓ 5 callersMethodclose_restart

src/io/persist.cpp:322

↓ 5 callersMethodcollect_background_data_fetch

src/data_coordinator/buffered_data_coordinator.cpp:250

↓ 5 callersMethodcompute_backward

include/lbann/utils/fftw_wrapper.hpp:337

← previous next →401–500 of 8,063, ranked by callers