MCPcopy Create free account

hub / github.com/NVIDIA-Merlin/HugeCTR / functions

Functions3,072 in github.com/NVIDIA-Merlin/HugeCTR

⨍ Functions 3,072◇ Types & classes 981↳ Endpoints 3
↓ 2 callersMethodon_eval_end
HugeCTR/include/pybind/training_callback.hpp:107
↓ 2 callersMethodon_eval_start
HugeCTR/include/pybind/training_callback.hpp:104
↓ 2 callersMethodon_training_start
HugeCTR/include/pybind/training_callback.hpp:98
↓ 2 callersMethodplan
(self)
samples/dlrm/sharding/planner.py:289
↓ 2 callersFunctionprint_array
test/utest/embedding_collection/embedding_collection_utils.hpp:28
↓ 2 callersFunctionprint_class_aucs
HugeCTR/src/pybind/model.cpp:678
↓ 2 callersMethodpush_bucket
( self, bucket_id: int, table_id: int, )
samples/dlrm/sharding/planner.py:115
↓ 2 callersMethodpush_bucket
( self, bucket_id: int, table_id: int, )
benchmarks/embedding_collection/hugectr/sharding/planner.py:242
↓ 2 callersMethodread_new_file
HugeCTR/src/data_readers/row_group_reading_thread.cpp:201
↓ 2 callersMethodready_to_collect
HugeCTR/src/data_readers/data_reader.cpp:196
↓ 2 callersMethodrecv_bcast
HugeCTR/include/collectives/ib_proxy.hpp:44
↓ 2 callersMethodreset_metadata
HugeCTR/src/data_readers/metadata.cpp:28
↓ 2 callersFunctionsave_table_to_filesystem_static
(var, optimizer, path, have_states)
sparse_operation_kit/sparse_operation_kit/dump_load.py:542
↓ 2 callersMethodscatter_update
(self, sparse_delta, use_locking=False, name=None)
sparse_operation_kit/sparse_operation_kit/dynamic_variable.py:327
↓ 2 callersMethodschedule_upload_here
HugeCTR/src/data_readers/multi_hot/detail/data_reader_impl.cpp:214
↓ 2 callersFunctionsegmented_reduce_sum
(d_in, num_sgements, begin_offsets, end_offsets)
notebooks/prototype_embedding_collection/op.py:30
↓ 2 callersMethodset_absolute_stream
HugeCTR/src/pipeline.cpp:39
↓ 2 callersMethodset_endpoint_override
HugeCTR/src/io/gcs_filesystem.cpp:78
↓ 2 callersMethodset_evaluate_layers
HugeCTR/src/core23_network.cpp:298
↓ 2 callersMethodset_evaluate_losses
HugeCTR/src/core23_network.cpp:316
↓ 2 callersMethodset_parallel
HugeCTR/embedding_storage/weight_io/data_info.hpp:100
↓ 2 callersMethodset_shape
HugeCTR/core23/tensor_params.hpp:106
↓ 2 callersMethodset_stream
HugeCTR/include/gpu_resource.hpp:74
↓ 2 callersMethodset_train_layers
HugeCTR/src/core23_network.cpp:282
↓ 2 callersMethodset_train_losses
HugeCTR/src/core23_network.cpp:311
↓ 2 callersMethodset_weight
@brief a modifier to reserve a weight tensor at idx with the specified dims. @details Usage: In a child class, this->set_weight(0, dims);
HugeCTR/include/trainable_layer.hpp:78
↓ 2 callersMethodset_wgrad
@brief a modifier to reserve a weight tensor at idx with the specified dims. @details Usage: In a child class, this->set_wgrad(0, dims);
HugeCTR/include/trainable_layer.hpp:98
↓ 2 callersFunctionsetup
(app)
docs/source/conf.py:194
↓ 2 callersMethodshrink
HugeCTR/include/tensor2.hpp:171
↓ 2 callersMethodskip_read
* skip data reading in read_a_batch() */
HugeCTR/include/data_readers/parquet_data_reader_worker.hpp:162
↓ 2 callersMethodstart
HugeCTR/include/data_readers/data_reader_worker_group.hpp:150
↓ 2 callersMethodstart_data_reading
HugeCTR/include/pybind/model.hpp:423
↓ 2 callersFunctionstep
(params, indices, use_filter=False)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup/lookup_sparse_hkv_low_frequency_test.py:79
↓ 2 callersFunctionstep2
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup/lookup_sparse_localized_test.py:145
↓ 2 callersFunctionstep2
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup_sp_weight/lookup_sparse_localized_test.py:144
↓ 2 callersMethodsubscribable
HugeCTR/core23/buffer.hpp:50
↓ 2 callersMethodswap
HugeCTR/core23/tensor.cpp:70
↓ 2 callersFunctiontest_impl
test/utest/core23/legacy_tensor_test.cpp:38
↓ 2 callersFunctiontest_impl
test/utest/core23/curand_generator_test.cpp:28
↓ 2 callersFunctiontime_double
HugeCTR/include/data_readers/multi_hot/detail/time_helper.hpp:20
↓ 2 callersFunctionto_df
(file_path)
samples/din/utils/1_convert_pd.py:21
↓ 2 callersFunctionto_df
(file_path)
samples/bst/utils/1_convert_pd.py:21
↓ 2 callersMethodtrain
(self, interval=1000, eval_interval=3793, eval_in_last=False, early_stop=-1, epochs=1)
sparse_operation_kit/documents/tutorials/DLRM_Benchmark/trainer.py:202
↓ 2 callersMethodtrain
(self, interval=10, eval_interval=3793, eval_in_last=False, early_stop=-1, epochs=1)
sparse_operation_kit/SOK_DLRM_Benchmark/trainer.py:214
↓ 2 callersFunctiontranspose
test/utest/core23_layer_test/matrix_multiply_test.cpp:54
↓ 2 callersFunctiontranspose_qkv_back_cpu
test/utest/core23_layer_test/multi_head_attention_layer_test.cpp:210
↓ 2 callersMethodtype
HugeCTR/core23/data_type.hpp:79
↓ 2 callersFunctionunaryAndGemv
test/utest/prims/reduce.h:121
↓ 2 callersMethodunfreeze
HugeCTR/include/embeddings/localized_slot_sparse_embedding_hash.hpp:605
↓ 2 callersMethodupdate
HugeCTR/include/embedding_training_cache/embedding_training_cache.hpp:60
↓ 2 callersMethodupdate
HugeCTR/embedding_storage/dynamic_embedding_cpu.hpp:230
↓ 2 callersMethodupdate
test/utest/embedding_collection/embedding_table_cpu.hpp:102
↓ 2 callersMethodupdate
test/utest/optimizer/optimizer_cpu.hpp:215
↓ 2 callersMethodupdate_mutable_data
HugeCTR/embedding/common.cpp:466
↓ 2 callersFunctionupload2hdfs_test
test/utest/io/hdfs_backend_test.cpp:87
↓ 2 callersMethoduse_gpu_learning_rate_scheduling
HugeCTR/include/pybind/model.hpp:479
↓ 2 callersMethodvalue
(self)
sparse_operation_kit/sparse_operation_kit/dynamic_variable.py:372
↓ 2 callersMethodwait_recv_bcast
HugeCTR/include/collectives/ib_proxy.hpp:45
↓ 2 callersMethodworld_rank
* Rank of this process in MPI world */
HugeCTR/core23/mpi_init_service.hpp:32
↓ 2 callersFunctionzeros_async
HugeCTR/core23/tensor_operations.cpp:44
↓ 2 callersFunctionzeros_sync
HugeCTR/core23/tensor_operations.cpp:39
↓ 1 callersMethodAssign
sparse_operation_kit/kit_src/variable/kernels/dummy_var.cc:73
↓ 1 callersFunctionAwsWriteableStreamFactory
HugeCTR/src/io/s3_filesystem.cpp:149
↓ 1 callersFunctionCommonPybind
HugeCTR/include/pybind/common_wrapper.hpp:31
↓ 1 callersFunctionConvertToBufferRequirements
HugeCTR/core23/details/tensor_helpers.cpp:49
↓ 1 callersMethodCopyEvictKeys
sparse_operation_kit/kit_src/variable/kernels/dummy_var.cc:97
↓ 1 callersFunctionDataGeneratorPybind
HugeCTR/include/pybind/data_generator_wrapper.hpp:27
↓ 1 callersFunctionDataReaderPybind
HugeCTR/include/pybind/data_reader_wrapper.hpp:27
↓ 1 callersFunctionDataSourcePybind
HugeCTR/include/pybind/data_source_wrapper.hpp:27
↓ 1 callersFunctionEmbeddingCollectionPybind
HugeCTR/include/pybind/embedding_collection_wrapper.hpp:28
↓ 1 callersMethodExport
sparse_operation_kit/kit_src/variable/kernels/dummy_var.cc:61
↓ 1 callersMethodExportIf
sparse_operation_kit/kit_src/variable/kernels/dummy_var.cc:67
↓ 1 callersFunctionGetDefaultCPUAllocator
HugeCTR/core23/allocator_factory.cpp:45
↓ 1 callersFunctionGetDefaultGPUAllocator
HugeCTR/core23/allocator_factory.cpp:30
↓ 1 callersFunctionGetDefaultUnifiedAllocator
HugeCTR/core23/allocator_factory.cpp:60
↓ 1 callersFunctionGetRandomBufferChannelName
HugeCTR/core23/buffer_channel_helpers.cpp:26
↓ 1 callersFunctionGetValidAlignment
HugeCTR/core23/details/tensor_helpers.cpp:28
↓ 1 callersFunctionGetWeightHalfBufferChannel
HugeCTR/src/network_buffer_channels.cpp:56
↓ 1 callersFunctionLearningRateSchedulerPybind
HugeCTR/include/pybind/learning_rate_scheduler_wrapper.hpp:27
↓ 1 callersFunctionModelPybind
HugeCTR/include/pybind/model_wrapper.hpp:28
↓ 1 callersFunctionOptimizerPybind
HugeCTR/include/pybind/optimizer_wrapper.hpp:48
↓ 1 callersMethodScatterAdd
sparse_operation_kit/kit_src/variable/kernels/dummy_var.cc:103
↓ 1 callersMethodScatterUpdate
sparse_operation_kit/kit_src/variable/kernels/dummy_var.cc:111
↓ 1 callersFunctionSolverPybind
HugeCTR/include/pybind/solver_wrapper.hpp:89
↓ 1 callersMethodSparseRead
sparse_operation_kit/kit_src/variable/kernels/dummy_var.cc:81
↓ 1 callersMethodSparseReadEvict
sparse_operation_kit/kit_src/variable/kernels/dummy_var.cc:89
↓ 1 callersMethodSparseTensor
HugeCTR/include/tensor2.hpp:266
↓ 1 callersFunctionToGoogleCloudOptions
* @brief Turn GCSConfigs into Google Cloud Options. * * @param GCSConfigs * @return google::cloud::Options */
HugeCTR/src/io/gcs_filesystem.cpp:113
↓ 1 callersFunctionTrainingCallbackPybind
HugeCTR/include/pybind/training_callback.hpp:112
↓ 1 callersFunctionZipf
Generate Zipf-like random variables, but in inclusive [min...max] interval
benchmarks/embedding_collection/dataset/180table_70B_hotness80.py:48
↓ 1 callersFunctionZipf
Generate Zipf-like random variables, but in inclusive [min...max] interval
benchmarks/embedding_collection/dataset/7table_470B_hotness20.py:40
↓ 1 callersFunctionZipf
Generate Zipf-like random variables, but in inclusive [min...max] interval
benchmarks/embedding_collection/dataset/200table_100B_hotness20.py:50
↓ 1 callersFunctionZipf
Generate Zipf-like random variables, but in inclusive [min...max] interval
benchmarks/embedding_collection/dataset/510table_110B_hotness5.py:40
↓ 1 callersFunction_GetSOKVersion
()
sparse_operation_kit/setup.py:31
↓ 1 callersFunction_GetSOKVersion
()
sparse_operation_kit/documents/source/conf.py:42
↓ 1 callersMethod__init__
(self, param: EmbeddingTableParam)
notebooks/prototype_embedding_collection/embedding_table.py:228
↓ 1 callersMethod__init__
(self)
sparse_operation_kit/documents/tutorials/DLRM/utils.py:24
↓ 1 callersMethod__init__
( self, initial_value=None, trainable=None, collections=None, validate
sparse_operation_kit/sparse_operation_kit/distributed_variable.py:236
↓ 1 callersMethod__init__
(self)
test/sok_perf_test/dlrm/utils.py:24
↓ 1 callersMethod__init__
( self, max_vocabulary_size_per_gpu, embedding_vec_size, slot_num, nnz
test/sok_perf_test/demo/model.py:148
← previousnext →501–600 of 3,072, ranked by callers