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
↓ 1 callersMethodset_buffer
HugeCTR/include/tensor2.hpp:185
↓ 1 callersFunctionset_comm_tool
Set the communication tool. Note that the communication tool cannot be set more than once.
sparse_operation_kit/sparse_operation_kit/communication.py:173
↓ 1 callersFunctionset_default_alloctor
sparse_operation_kit/kit_src/lookup/kernels/embedding_collection.cc:44
↓ 1 callersMethodset_gpu_barrier
HugeCTR/embedding/hier_model_parallel_embedding.hpp:114
↓ 1 callersMethodset_learning_rate_scheduler
HugeCTR/include/optimizer.hpp:264
↓ 1 callersMethodset_optimizer
HugeCTR/src/core23_network.cpp:332
↓ 1 callersMethodset_raw_metrics
HugeCTR/src/core23_network.cpp:328
↓ 1 callersMethodset_ready_to_transfer
HugeCTR/include/collectives/collective.hpp:46
↓ 1 callersMethodset_resource_manager
HugeCTR/include/data_readers/data_reader_worker_group.hpp:146
↓ 1 callersMethodset_stop
(self, message)
sparse_operation_kit/documents/tutorials/DLRM/utils.py:27
↓ 1 callersMethodset_stop
(self, message)
test/sok_perf_test/dlrm/utils.py:27
↓ 1 callersMethodset_top_and_bottom_layers
HugeCTR/src/core23_network.cpp:322
↓ 1 callersMethodset_user_name
HugeCTR/src/io/hadoop_filesystem.cpp:33
↓ 1 callersFunctionsetup_rmm_pool
(client, pool_size)
samples/bst/utils/4_nvt_process.py:56
↓ 1 callersFunctionsgd_update_grad
* SGD (Stateless) * --------------- * g_i = -eta * g_i / s */
HugeCTR/embedding_storage/optimizers.hpp:28
↓ 1 callersMethodshould_stop
(self)
sparse_operation_kit/documents/tutorials/DLRM/utils.py:35
↓ 1 callersMethodshould_stop
(self)
test/sok_perf_test/dlrm/utils.py:35
↓ 1 callersFunctionsimple_read_write_test
test/utest/io/hdfs_backend_test.cpp:53
↓ 1 callersFunctionsimple_read_write_test
test/utest/io/local_fs_test.cpp:40
↓ 1 callersFunctionsimple_read_write_test_with_builder
test/utest/io/hdfs_backend_test.cpp:41
↓ 1 callersFunctionsimple_read_write_test_with_builder
test/utest/io/local_fs_test.cpp:28
↓ 1 callersFunctionsklearn_ref
test/utest/metrics/auc_test.cpp:36
↓ 1 callersFunctionsok_step
(indices, weight, var)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/variable/dynamic_variable_test.py:49
↓ 1 callersFunctionsok_step
(indices, weight, var)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/variable/sok_sgd_test.py:41
↓ 1 callersFunctionsok_step
(indices, weight, var)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/variable/hkv_test.py:52
↓ 1 callersFunctionsok_step_eager
(indices, weight, var)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/variable/dynamic_variable_test.py:89
↓ 1 callersFunctionsok_step_eager
(indices, weight, var)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/variable/sok_sgd_test.py:79
↓ 1 callersFunctionsok_step_eager
(indices, weight, var)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/variable/hkv_test.py:92
↓ 1 callersFunctionsort_lookup_ids
HugeCTR/embedding/common.cpp:500
↓ 1 callersMethodsource_available
HugeCTR/src/data_readers/row_group_reading_thread.cpp:144
↓ 1 callersMethodsparse_read
(self, indices, name=None)
sparse_operation_kit/sparse_operation_kit/dynamic_variable.py:294
↓ 1 callersFunctionsplit_column_wise_sharding_config
HugeCTR/src/embeddings/embedding_collection.cpp:25
↓ 1 callersFunctionsplit_data
(filename, split_num, save_prefix)
sparse_operation_kit/documents/tutorials/DenseDemo/split_data.py:26
↓ 1 callersFunctionsplit_emb_and_dense_variables
(variables)
sparse_operation_kit/documents/tutorials/DenseDemo/run_tf.py:29
↓ 1 callersMethodsplit_hot_shard
split the shard with the largest hotness
samples/dlrm/sharding/planner.py:46
↓ 1 callersMethodsplit_oom_shard
(self, table_id: int)
samples/dlrm/sharding/planner.py:82
↓ 1 callersFunctionsplit_str
* @brief Split the path string by separator * * @param str * @param sep * @return std::vector<std::string> */
HugeCTR/include/io/s3_utils.hpp:51
↓ 1 callersMethodstart
HugeCTR/src/data_readers/data_collector.cpp:37
↓ 1 callersFunctionstep
(params, indices)
sparse_operation_kit/example/sok_incremental_dump/incremental_dump_test.py:84
↓ 1 callersFunctionstep
(params, indices)
sparse_operation_kit/example/sok_dump_load/dump_load.py:52
↓ 1 callersFunctionstep
()
sparse_operation_kit/sparse_operation_kit/test/op_test/dummy_var/dummy_var_handle_test.py:33
↓ 1 callersFunctionstep
(param, indices, optimizer)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup/all2all_dense_embedding_test.py:70
↓ 1 callersFunctionstep
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup/lookup_sparse_distributed_dynamic_test.py:98
↓ 1 callersFunctionstep
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup/lookup_sparse_localized_hkv_test.py:123
↓ 1 callersFunctionstep
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup/lookup_sparse_distributed_hkv_test.py:123
↓ 1 callersFunctionstep
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup/lookup_sparse_localized_dynamic_test.py:119
↓ 1 callersFunctionstep
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup/lookup_sparse_hkv_incremental_dump_test.py:126
↓ 1 callersFunctionstep
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup/lookup_sparse_localized_test.py:92
↓ 1 callersFunctionstep
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup/lookup_sparse_distributed_test.py:93
↓ 1 callersFunctionstep
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup_sp_weight/lookup_sparse_distributed_dynamic_test.py:116
↓ 1 callersFunctionstep
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup_sp_weight/lookup_sparse_localized_dynamic_test.py:119
↓ 1 callersFunctionstep
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup_sp_weight/lookup_sparse_localized_test.py:93
↓ 1 callersFunctionstep
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup_sp_weight/lookup_sparse_distributed_test.py:107
↓ 1 callersFunctionstep
(param, indices, optimizer)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup/all2all_dense_embedding_test.py:66
↓ 1 callersFunctionstep
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup/lookup_sparse_distributed_dynamic_test.py:86
↓ 1 callersFunctionstep
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup/lookup_sparse_localized_hkv_test.py:101
↓ 1 callersFunctionstep
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup/lookup_sparse_distributed_hkv_test.py:87
↓ 1 callersFunctionstep
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup/lookup_sparse_distributed_dynamic_with_sample_key_zero_test.py:73
↓ 1 callersFunctionstep
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup/lookup_sparse_localized_dynamic_test.py:99
↓ 1 callersFunctionstep
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup/lookup_sparse_distributed_dynamic_test_without_redefine.py:94
↓ 1 callersFunctionstep
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup/lookup_sparse_hkv_incremental_dump_test.py:90
↓ 1 callersFunctionstep
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup/lookup_sparse_localized_test.py:94
↓ 1 callersFunctionstep
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup/lookup_sparse_distributed_test.py:82
↓ 1 callersFunctionstep
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/dump_load/dump_load_localized_dynamic.py:68
↓ 1 callersFunctionstep
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/dump_load/dump_load_localized_static.py:66
↓ 1 callersFunctionstep
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/dump_load/dump_load_distribute_dynamic.py:63
↓ 1 callersFunctionstep
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/dump_load/dump_load_distribute_static_big_table.py:50
↓ 1 callersFunctionstep
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/dump_load/dump_load_distribute_static.py:62
↓ 1 callersFunctionstep
(params, indices, sp_weights)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup_sp_weight/lookup_sparse_distributed_dynamic_test.py:79
↓ 1 callersFunctionstep
(params, indices, sp_weights)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup_sp_weight/lookup_sparse_localized_dynamic_test.py:97
↓ 1 callersFunctionstep
(params, indices, sp_weights)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup_sp_weight/lookup_sparse_localized_test.py:98
↓ 1 callersFunctionstep
(params, indices, sp_weights)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup_sp_weight/lookup_sparse_distributed_test.py:83
↓ 1 callersFunctionstep2
(param, indices, optimizer)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup/all2all_dense_embedding_test.py:96
↓ 1 callersFunctionstep2
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup/lookup_sparse_distributed_dynamic_test.py:150
↓ 1 callersFunctionstep2
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup/lookup_sparse_localized_hkv_test.py:176
↓ 1 callersFunctionstep2
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup/lookup_sparse_distributed_hkv_test.py:176
↓ 1 callersFunctionstep2
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup/lookup_sparse_localized_dynamic_test.py:172
↓ 1 callersFunctionstep2
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup/lookup_sparse_distributed_test.py:141
↓ 1 callersFunctionstep2
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup_sp_weight/lookup_sparse_distributed_dynamic_test.py:174
↓ 1 callersFunctionstep2
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup_sp_weight/lookup_sparse_localized_dynamic_test.py:176
↓ 1 callersFunctionstep2
(params)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf1/lookup_sp_weight/lookup_sparse_distributed_test.py:161
↓ 1 callersFunctionstep2
(param, indices, optimizer)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup/all2all_dense_embedding_test.py:90
↓ 1 callersFunctionstep2
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup/lookup_sparse_distributed_dynamic_test.py:116
↓ 1 callersFunctionstep2
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup/lookup_sparse_localized_hkv_test.py:134
↓ 1 callersFunctionstep2
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup/lookup_sparse_distributed_hkv_test.py:116
↓ 1 callersFunctionstep2
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup/lookup_sparse_localized_dynamic_test.py:132
↓ 1 callersFunctionstep2
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup/lookup_sparse_distributed_dynamic_test_without_redefine.py:124
↓ 1 callersFunctionstep2
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup/lookup_sparse_localized_test.py:122
↓ 1 callersFunctionstep2
(params, indices)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup/lookup_sparse_distributed_test.py:110
↓ 1 callersFunctionstep2
(params, indices, sp_weights)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup_sp_weight/lookup_sparse_distributed_dynamic_test.py:115
↓ 1 callersFunctionstep2
(params, indices, sp_weights)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup_sp_weight/lookup_sparse_localized_dynamic_test.py:136
↓ 1 callersFunctionstep2
(params, indices, sp_weights)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup_sp_weight/lookup_sparse_localized_test.py:133
↓ 1 callersFunctionstep2
(params, indices, sp_weights)
sparse_operation_kit/sparse_operation_kit/test/function_test/tf2/lookup_sp_weight/lookup_sparse_distributed_test.py:117
↓ 1 callersMethodstop
HugeCTR/src/data_readers/row_group_reading_thread.cpp:122
↓ 1 callersMethodstream_wait_dense_tensors
HugeCTR/src/data_readers/multi_hot/async_data_reader.cpp:394
↓ 1 callersFunctionsub_cpu
test/utest/core23_layer_test/sub_layer_test.cpp:31
↓ 1 callersFunctionsub_dgrad_cpu
test/utest/core23_layer_test/sub_layer_test.cpp:36
↓ 1 callersMethodsubmit
HugeCTR/src/data_readers/multi_hot/detail/aio_context.cpp:52
↓ 1 callersFunctionsum_ex_cpu
test/utest/core23_layer_test/masked_softmax_layer_test.cpp:59
↓ 1 callersFunctionsum_ex_cpu
test/utest/core23_layer_test/softmax_layer_test.cpp:41
← previousnext →1,001–1,100 of 3,072, ranked by callers