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 callersFunctioncreate_compression_param_from_ebc_config
HugeCTR/include/embeddings/embedding_collection.hpp:314
↓ 1 callersFunctioncreate_data_reader
( data_reader_type, source, eval_source, num_samples, eval_num_samples, float_label_de
test/pybind_test/model_test.py:181
↓ 1 callersMethodcreate_data_reader_threads
* Create threads to run data reader workers */
HugeCTR/include/data_readers/data_reader_worker_group.hpp:105
↓ 1 callersMethodcreate_dir
HugeCTR/src/io/s3_filesystem.cpp:178
↓ 1 callersMethodcreate_embedding
(self)
notebooks/prototype_embedding_collection/embedding.py:488
↓ 1 callersMethodcreate_embedding_collection
(self, gpu_id)
notebooks/prototype_embedding_collection/embedding.py:89
↓ 1 callersMethodcreate_graph
(self, name="hugectr_graph")
onnx_converter/hugectr2onnx/graph_builder.py:1702
↓ 1 callersFunctioncreate_grouped_embedding_param_from_ebc_config
HugeCTR/include/embeddings/embedding_collection.hpp:280
↓ 1 callersFunctioncreate_grouped_embedding_tables
HugeCTR/embedding_storage/embedding_table.cpp:23
↓ 1 callersFunctioncreate_lookup_params_from_ebc_config
HugeCTR/include/embeddings/embedding_collection.hpp:228
↓ 1 callersFunctioncreate_optimizer
( optimizer_type, beta1, beta2, epsilon, update_type, momentum_factor, atomic_upda
test/pybind_test/model_test.py:203
↓ 1 callersFunctioncreate_shard_matrix_from_ebc_config
HugeCTR/include/embeddings/embedding_collection.hpp:256
↓ 1 callersFunctioncreate_solver
( max_eval_batches: int, batchsize_eval: int, batchsize: int, vvgpu: List[List[int]], lr:
test/pybind_test/model_test.py:148
↓ 1 callersMethodcreate_static_embedding_table
(gpu_id, embedding_collection_param: EmbeddingCollectionParam, embedding_table_param_list: typing.List[Embeddi
notebooks/prototype_embedding_collection/embedding_table.py:277
↓ 1 callersFunctioncreate_table_name_to_id_dict_from_ebc_config
HugeCTR/include/embeddings/embedding_collection.hpp:205
↓ 1 callersFunctioncreate_table_params_from_ebc_config
HugeCTR/include/embeddings/embedding_collection.hpp:242
↓ 1 callersFunctioncriteo
( model: hugectr.Model, optimizer: hugectr.Optimizer_t, single_slot: bool, )
test/pybind_test/model_test.py:225
↓ 1 callersMethoddata
HugeCTR/core23/tensor.hpp:60
↓ 1 callersMethoddata
HugeCTR/core23/details/tensor_impl.cpp:39
↓ 1 callersFunctiondcnv1
( model: hugectr.Model, optimizer: hugectr.Optimizer_t, )
test/pybind_test/model_test.py:322
↓ 1 callersFunctiondcnv2
( model: hugectr.Model, optimizer: hugectr.Optimizer_t, )
test/pybind_test/model_test.py:436
↓ 1 callersMethoddeallocate
HugeCTR/include/general_buffer2.hpp:29
↓ 1 callersMethoddecay
HugeCTR/core23/buffer.hpp:57
↓ 1 callersMethoddeduct_mem_cap_for_dp
( self, dp_table_id: list, )
samples/dlrm/sharding/planner.py:196
↓ 1 callersFunctiondeduplicate_sorted_table_ids
HugeCTR/embedding/common.cpp:521
↓ 1 callersFunctiondeepfm
( model: hugectr.Model, optimizer: hugectr.Optimizer_t, )
test/pybind_test/model_test.py:685
↓ 1 callersMethoddelete_dir
HugeCTR/embedding_storage/weight_io/fs_interface.cpp:64
↓ 1 callersMethoddequeue
HugeCTR/include/data_readers/multi_hot/detail/work_queue.hpp:68
↓ 1 callersFunctiondice_bprop
test/utest/core23_layer_test/prelu_dice_layer_test.cpp:73
↓ 1 callersFunctiondice_fprop
test/utest/core23_layer_test/prelu_dice_layer_test.cpp:61
↓ 1 callersMethoddo_h2d
HugeCTR/src/data_readers/parquet_data_reader_worker.cpp:23
↓ 1 callersMethoddownload_opt_states_to_host
HugeCTR/src/core23_network.cpp:130
↓ 1 callersMethoddownload_params_to_host
HugeCTR/src/core23_network.cpp:117
↓ 1 callersMethoddump
(self)
notebooks/prototype_embedding_collection/embedding_table.py:65
↓ 1 callersMethoddump_model
(self)
sparse_operation_kit/example/sok_train_hps_inference/sok_train.py:319
↓ 1 callersFunctiondump_per_table
(var, optimizer, path, have_states)
sparse_operation_kit/sparse_operation_kit/dump_load.py:1069
↓ 1 callersFunctiondump_table
(path, dump_vars, optimizer)
sparse_operation_kit/sparse_operation_kit/dump_load.py:1136
↓ 1 callersFunctionelementwise_multiply_cpu
test/utest/core23_layer_test/elementwise_multiply_layer_test.cpp:42
↓ 1 callersFunctionelementwise_multiply_dgrad_cpu
test/utest/core23_layer_test/elementwise_multiply_layer_test.cpp:55
↓ 1 callersFunctionelu_bprop_cpu
test/utest/core23_layer_test/elu_layer_test.cpp:53
↓ 1 callersFunctionelu_cpu
test/utest/core23_layer_test/elu_layer_test.cpp:46
↓ 1 callersMethodembedding_backward_cpu
test/utest/embedding_collection/reference_embedding.hpp:967
↓ 1 callersMethodembedding_dump
(self, path, opt)
sparse_operation_kit/example/sok_train_hps_inference/sok_train.py:255
↓ 1 callersMethodembedding_dump
same with model.embedding_dump function
test/utest/embedding_collection/test_embedding_collection_load_dump.cpp:163
↓ 1 callersMethodembedding_forward_cpu
test/utest/embedding_collection/reference_embedding.hpp:952
↓ 1 callersMethodembedding_layer
(self)
test/sok_perf_test/dlrm/models.py:292
↓ 1 callersMethodembedding_load
same with model.embedding_load function
test/utest/embedding_collection/test_embedding_collection_load_dump.cpp:51
↓ 1 callersMethodembedding_update_cpu
test/utest/embedding_collection/reference_embedding.hpp:237
↓ 1 callersMethodenabled_at
HugeCTR/core23/logger.hpp:280
↓ 1 callersMethodenqueue
HugeCTR/include/data_readers/multi_hot/detail/work_queue.hpp:44
↓ 1 callersFunctionex_cpu
test/utest/core23_layer_test/masked_softmax_layer_test.cpp:71
↓ 1 callersFunctionex_cpu
test/utest/core23_layer_test/softmax_layer_test.cpp:55
↓ 1 callersFunctionex_cpu
test/utest/core23_layer_test/multi_head_attention_layer_test.cpp:43
↓ 1 callersMethodexchange_wgrad
HugeCTR/src/core23_network.cpp:216
↓ 1 callersMethodexchange_wgrad
HugeCTR/src/pybind/model_pipeline.cpp:30
↓ 1 callersMethodexec
HugeCTR/src/graph_wrapper.cpp:37
↓ 1 callersMethodexecute
HugeCTR/include/data_readers/multi_hot/detail/device_transfer.hpp:32
↓ 1 callersMethodfill
HugeCTR/include/data_simulator.hpp:175
↓ 1 callersFunctionfilter_dense_tables
(candidate_table_ids, threshold)
benchmarks/embedding_collection/hugectr/sharding/generate_plan.py:274
↓ 1 callersFunctionfilter_dp_lookup_ids
HugeCTR/embedding/common.cpp:199
↓ 1 callersFunctionfilter_dp_tables
(candidate_table_ids, threshold)
benchmarks/embedding_collection/hugectr/sharding/generate_plan.py:240
↓ 1 callersFunctionfilter_mp_dense_lookup_ids
HugeCTR/embedding/common.cpp:237
↓ 1 callersFunctionfilter_mp_sparse_lookup_ids
HugeCTR/embedding/common.cpp:220
↓ 1 callersMethodfinalize_batch
HugeCTR/src/data_readers/data_collector.cpp:184
↓ 1 callersMethodfind_next_file_and_group
HugeCTR/src/data_readers/file_source_parquet.cpp:70
↓ 1 callersFunctionfm_order2_bprop_cpu
test/utest/core23_layer_test/fm_order2_layer_test.cpp:82
↓ 1 callersFunctionfm_order2_fprop_cpu
test/utest/core23_layer_test/fm_order2_layer_test.cpp:44
↓ 1 callersMethodforward_communication_per_gpu
(self)
notebooks/prototype_embedding_collection/embedding.py:272
↓ 1 callersMethodforward_per_gpu
(self, key, bucket_range, embedding_table, output_buffer)
notebooks/prototype_embedding_collection/embedding.py:414
↓ 1 callersFunctionforward_void_pointer
HugeCTR/include/tensor2.hpp:34
↓ 1 callersMethodfprop
HugeCTR/include/embeddings/all2all_hierarchical.hpp:191
↓ 1 callersFunctionftrl_update_grad
* FTRL * ---- * g_i = g_i / s * sigma = (sqrt(n_i + g_i^2) - sqrt(n_i)) / eta * n_i = n_i + g_i^2 * z_i = z_i + g_i - sigma w_i * * if abs(z_i)
HugeCTR/embedding_storage/optimizers.hpp:174
↓ 1 callersFunctionfully_connected_layer_test_3d
test/utest/core23_layer_test/fully_connected_layer_test.cpp:162
↓ 1 callersFunctiongcs_configs_test
test/utest/io/gcs_backend_test.cpp:38
↓ 1 callersFunctiongcs_path_match_test
test/utest/io/gcs_backend_test.cpp:28
↓ 1 callersFunctiongcs_path_test
test/utest/io/gcs_backend_test.cpp:45
↓ 1 callersFunctiongcs_simple_read_write_test_with_dsp
test/utest/io/gcs_backend_test.cpp:58
↓ 1 callersFunctiongcs_simple_read_write_test_with_path
test/utest/io/gcs_backend_test.cpp:71
↓ 1 callersFunctiongen_neg
()
samples/din/utils/3_padding.py:37
↓ 1 callersFunctiongen_neg
()
samples/bst/utils/3_padding.py:37
↓ 1 callersMethodgenerate
HugeCTR/src/data_generator.cpp:90
↓ 1 callersFunctiongenerate_datas
(args)
sparse_operation_kit/documents/tutorials/DenseDemo/gen_data.py:25
↓ 1 callersMethodgenerate_embedding_plan
(self, strategy)
notebooks/prototype_embedding_collection/embedding.py:40
↓ 1 callersMethodgenerate_embedding_plan_from_json_file
(self, plan_file)
notebooks/prototype_embedding_collection/embedding.py:14
↓ 1 callersFunctiongenerate_kv_file_for_hps
(args)
sparse_operation_kit/example/sok_train_hps_inference/sok_train.py:330
↓ 1 callersFunctiongenerate_plan_ragged_ev_size
( table_id_list: List[int], slot_size_array: List[int], multi_hot_sizes: List[int], ev_size_li
benchmarks/embedding_collection/hugectr/sharding/generate_plan.py:57
↓ 1 callersFunctiongenerate_ragged_tensor_samples
( embedding_table_sizes, batch_size, lookup_num, hotness, iters )
sparse_operation_kit/example/sok_train_hps_inference/sok_train.py:53
↓ 1 callersFunctiongenerate_random_samples
(num_samples, vocabulary_range_per_slot, max_nnz, dense_dim)
sparse_operation_kit/example/sok_train_hps_inference/hps_use_sok_weight_inference.py:59
↓ 1 callersFunctiongenerate_shard_plan
(slot_size_array, num_gpus)
samples/ftrl/dlrm_train_ftrl.py:72
↓ 1 callersFunctiongenerate_shard_plan
(slot_size_array, num_gpus)
test/embedding_collection_test/dlrm_train_ftrl.py:76
↓ 1 callersFunctiongenerate_sparse_keys
( num_samples, vocabulary_range_per_slot, max_nnz, key_dtype=args["np_key_type"] )
sparse_operation_kit/example/sok_train_hps_inference/hps_use_sok_weight_inference.py:60
↓ 1 callersFunctionget_accum
test/utest/core23_layer_test/interaction_layer_test.cpp:49
↓ 1 callersMethodget_buffer
HugeCTR/embedding_storage/weight_io/data_info.hpp:116
↓ 1 callersMethodget_cat_names
HugeCTR/include/data_readers/metadata.hpp:79
↓ 1 callersFunctionget_categorical_feature_type
(size: int)
sparse_operation_kit/documents/tutorials/DLRM_Benchmark/preprocess/parquet_to_binary.py:44
↓ 1 callersMethodget_cc_minor
HugeCTR/include/gpu_resource.hpp:99
↓ 1 callersFunctionget_cmake_args
()
sparse_operation_kit/setup.py:44
↓ 1 callersFunctionget_core_resource_managers
test/utest/data_distributor/data_distributor_tests.cpp:64
↓ 1 callersMethodget_current_file_size
* @brief Get the current file size * * @return the file size */
HugeCTR/include/io/file_loader.hpp:77
↓ 1 callersFunctionget_dense_layer_from_json
HugeCTR/src/pybind/add_dense_layer.cpp:432
← previousnext →701–800 of 3,072, ranked by callers