MCPcopy Create free account

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

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

↓ 7 callersMethodwait
HugeCTR/include/data_readers/multi_hot/detail/system_latch.hpp:57
↓ 6 callersFunctionGetNetworkBufferChannel
HugeCTR/src/network_buffer_channels.cpp:41
↓ 6 callersFunctioncompare_array_approx
(results, reference, model_name, abs_th, rel_th)
test/onnx_converter_test/utils.py:84
↓ 6 callersFunctioncpu_add_bias
test/utest/core23_layer_test/gru_layer_test.cpp:90
↓ 6 callersFunctioncpu_mm
test/utest/core23_layer_test/fully_connected_layer_test.cpp:41
↓ 6 callersMethodcreate_drwg_parquet
HugeCTR/src/data_readers/data_reader.cpp:272
↓ 6 callersFunctiondata_reader_group_iter_squential_batch_test_impl
test/utest/data_reader/data_reader_parquet_test.cpp:490
↓ 6 callersMethoddrop
HugeCTR/embedding_storage/dynamic_embedding_cpu.hpp:49
↓ 6 callersFunctionfile_exist
* Check if file exist. */
HugeCTR/include/data_generator.hpp:42
↓ 6 callersMethodget_a_file_with_id
* Get a file name from the list. * @return the file name and id. */
HugeCTR/include/data_readers/file_list.hpp:105
↓ 6 callersMethodget_ar_comm
HugeCTR/include/collectives/collective.hpp:51
↓ 6 callersMethodget_gpu_global_id_from_local_id
HugeCTR/core/hctr_impl/hctr_backend.hpp:76
↓ 6 callersFunctionget_meta_info_offset
(variable_type, table_num)
sparse_operation_kit/sparse_operation_kit/dump_load.py:197
↓ 6 callersFunctionget_sok_optimizer_name
(optimizer)
sparse_operation_kit/sparse_operation_kit/dump_load.py:126
↓ 6 callersMethodhas_bucket_and_key
HugeCTR/include/io/s3_utils.hpp:127
↓ 6 callersFunctionloss_with_regularizer_test
test/utest/loss/loss_with_regularizer_test.cpp:86
↓ 6 callersFunctiontf_dataset
(sparse_keys, dense_features, labels, batchsize)
sparse_operation_kit/example/sok_train_hps_inference/sok_train.py:76
↓ 6 callersFunctionvec_to_string
HugeCTR/include/utils.hpp:138
↓ 5 callersMethodAsyncParam
HugeCTR/include/common.hpp:170
↓ 5 callersMethodEmbeddingCollectionConfig
if we need more configuration about EmbeddingCollection
HugeCTR/include/embeddings/embedding_collection.hpp:123
↓ 5 callersMethodEmbeddingTableConfig
HugeCTR/include/embeddings/embedding_collection.hpp:38
↓ 5 callersFunctionGetAllocator
HugeCTR/core23/allocator_factory.cpp:71
↓ 5 callersMethod__init__
Initializes the MLP layer. Args: units: Sequential list of layer sizes. List[int]. use_bias: Whether to include a bias term.
sparse_operation_kit/SOK_DLRM_Benchmark/model.py:88
↓ 5 callersMethodassign
HugeCTR/embedding_storage/dynamic_embedding_cpu.hpp:210
↓ 5 callersMethodcols
sparse_operation_kit/kit_src/variable/kernels/dummy_var.cc:55
↓ 5 callersMethodcurrent_batch_incomplete
HugeCTR/src/data_readers/data_reader.cpp:222
↓ 5 callersMethoddeallocate
HugeCTR/core23/details/pool_cuda_allocator.cpp:73
↓ 5 callersFunctionev_size_compensation
(ev_sizes_array)
benchmarks/embedding_collection/hugectr/sharding/planner.py:23
↓ 5 callersFunctiongenerate_parquet_input_files
generate parquet files and return data in the form of `row-major` stored in vector<>
test/utest/data_reader/data_reader_parquet_test.cpp:89
↓ 5 callersMethodget_current_batchsize_per_device
HugeCTR/src/data_readers/data_reader.cpp:201
↓ 5 callersMethodget_eval_metrics
HugeCTR/src/pybind/model.cpp:1196
↓ 5 callersMethodget_local_gpu_id
HugeCTR/core/hctr_impl/hctr_backend.hpp:66
↓ 5 callersFunctionget_nccl_dtype_from_tensor_scalar_type_core23
HugeCTR/embedding/operators/communication.cpp:24
↓ 5 callersMethodgreedy_plan
This is a heuristic based on greedy policy. The shard is placed to the bucket with the lowest hotness cost
benchmarks/embedding_collection/hugectr/sharding/planner.py:452
↓ 5 callersMethodpush
HugeCTR/include/data_readers/multi_hot/detail/work_queue.hpp:119
↓ 5 callersMethodscope
(self)
sparse_operation_kit/documents/tutorials/DLRM/utils.py:138
↓ 5 callersMethodset_label_weight
HugeCTR/include/loss.hpp:121
↓ 5 callersMethodset_source
HugeCTR/src/pybind/model.cpp:667
↓ 5 callersMethodtrain
HugeCTR/src/pybind/model.cpp:1048
↓ 4 callersFunctionGetBuffer
HugeCTR/core23/buffer_factory.cpp:38
↓ 4 callersFunctionGetWgradBufferChannel
HugeCTR/src/network_buffer_channels.cpp:60
↓ 4 callersFunctionGetWgradHalfBufferChannel
HugeCTR/src/network_buffer_channels.cpp:64
↓ 4 callersFunctionTEST
test/utest/resource_manager/resource_manager_test.cpp:17
↓ 4 callersMethod__init__
Initializes the MLP layer. Args: units: Sequential list of layer sizes. List[int]. use_bias: Whether to include a bias term.
sparse_operation_kit/documents/tutorials/DLRM_Benchmark/model.py:103
↓ 4 callersMethod__init__
( self, global_batch_size, vocabulary_size, slot_num, nnz_per_slot,
sparse_operation_kit/documents/tutorials/DenseDemo/models.py:199
↓ 4 callersMethod__init__
Initializes the MLP layer. Args: units: Sequential list of layer sizes. use_bias: Whether to include a bias term. acti
sparse_operation_kit/documents/tutorials/DLRM/models.py:123
↓ 4 callersMethod__init__
Initializes the MLP layer. Args: units: Sequential list of layer sizes. use_bias: Whether to include a bias term. acti
test/sok_perf_test/dlrm/models.py:125
↓ 4 callersMethod_create_slots
(self, vars)
sparse_operation_kit/sparse_operation_kit/optimizer.py:119
↓ 4 callersMethod_var_key
(self, var)
sparse_operation_kit/sparse_operation_kit/optimizer.py:114
↓ 4 callersMethod_var_key
(self, var)
sparse_operation_kit/sparse_operation_kit/optimizer.py:290
↓ 4 callersFunctionassign
Abbreviated as ``sok.assign``. Assign the indices and value tensor to the target variable. Parameters ---------- var: sok.Dynam
sparse_operation_kit/sparse_operation_kit/dynamic_variable.py:494
↓ 4 callersMethodcheck_overflow
HugeCTR/src/pybind/model.cpp:1251
↓ 4 callersFunctioncompare_array
test/utest/core23_layer_test/fully_connected_layer_half_test.cpp:59
↓ 4 callersFunctionconvert_bytes_to_int_list
(bytes_list, data_length, signed=False)
sparse_operation_kit/sparse_operation_kit/dump_load.py:252
↓ 4 callersMethodcount
HugeCTR/include/data_readers/multi_hot/detail/batch_locations.hpp:122
↓ 4 callersFunctioncpu_gpu_cmp
test/utest/test_utils.hpp:156
↓ 4 callersFunctiondeactivate_tensor
HugeCTR/include/parser.hpp:310
↓ 4 callersFunctiondump
Abbreviated as ``sok.dump``. Dump the embedding tables into one folder , will generate key and value file per table. Table name prefix
sparse_operation_kit/sparse_operation_kit/dump_load.py:1176
↓ 4 callersMethodempty
HugeCTR/core23/tensor_container.hpp:138
↓ 4 callersMethodend
HugeCTR/include/data_readers/multi_hot/detail/batch_locations.hpp:120
↓ 4 callersFunctionfill_test_impl
test/utest/core23/low_level_primitive_test.cpp:36
↓ 4 callersFunctionfully_connected_layer_test
test/utest/core23_layer_test/fully_connected_layer_half_test.cpp:69
↓ 4 callersFunctiongenerate_random_samples
(batch_size, iters, vocabulary_range_per_slot, max_nnz, dense_dim)
sparse_operation_kit/example/sok_train_hps_inference/sok_train.py:50
↓ 4 callersMethodget
HugeCTR/embedding_storage/dynamic_embedding_cpu.hpp:67
↓ 4 callersMethodget_current_loss
HugeCTR/src/pybind/model.cpp:1204
↓ 4 callersFunctionget_layer_names
HugeCTR/include/parser.hpp:320
↓ 4 callersMethodget_master_weights
HugeCTR/include/trainable_layer.hpp:156
↓ 4 callersMethodget_next
HugeCTR/include/learning_rate_scheduler.hpp:66
↓ 4 callersMethodget_num_elements
HugeCTR/include/tensor2.hpp:179
↓ 4 callersFunctionget_save_rounds
(tensor_shape, tensor_data_size)
sparse_operation_kit/sparse_operation_kit/dump_load.py:397
↓ 4 callersMethodget_size
HugeCTR/include/embedding_training_cache/parameter_server_manager.hpp:44
↓ 4 callersMethodget_sm_count
HugeCTR/include/gpu_resource.hpp:96
↓ 4 callersMethodget_value_tensor
HugeCTR/include/sparse_tensor.hpp:111
↓ 4 callersFunctionget_wgrad_ev_size
HugeCTR/embedding/common.cpp:622
↓ 4 callersFunctionhugectr2onnx_dcn_test
( batch_size, num_batches, data_file, graph_config, dense_model, sparse_models, on
test/onnx_converter_test/hugectr2onnx_dcn_test.py:24
↓ 4 callersMethodinit_data
HugeCTR/embedding/common.cpp:656
↓ 4 callersMethodis_open
HugeCTR/include/data_readers/checker.hpp:44
↓ 4 callersMethodload_by_id
HugeCTR/embedding_storage/dynamic_embedding_cpu.hpp:403
↓ 4 callersMethodload_embedding_weight
HugeCTR/embedding_storage/weight_io/parameter_IO.cpp:473
↓ 4 callersMethodlocal_reduce
HugeCTR/embedding/data_parallel_embedding.cpp:224
↓ 4 callersMethodlookup
HugeCTR/embedding_storage/dynamic_embedding_cpu.hpp:171
↓ 4 callersMethodmake_dir
HugeCTR/embedding_storage/weight_io/fs_interface.cpp:58
↓ 4 callersFunctionmatmul_cpu
test/utest/core23_layer_test/matrix_multiply_test.cpp:39
↓ 4 callersMethodprop_layers
HugeCTR/src/core23_network.cpp:375
↓ 4 callersFunctionregularizer_test_common
test/utest/regularizers/regularizer_test_common.cpp:44
↓ 4 callersFunctionrow_wise_sharding
test/utest/embedding_collection/configuration.hpp:225
↓ 4 callersMethodset_connection_configs
HugeCTR/src/io/hadoop_filesystem.cpp:35
↓ 4 callersMethodset_this_producer_status
HugeCTR/src/data_readers/row_group_reading_thread.cpp:162
↓ 4 callersMethodsubscribe
HugeCTR/core23/buffer.cpp:34
↓ 4 callersFunctiontable_wise_sharding
test/utest/embedding_collection/configuration.hpp:190
↓ 4 callersMethodto_dynamic
(self)
sparse_operation_kit/sparse_operation_kit/dynamic_variable.py:214
↓ 4 callersFunctionto_list
(any_obj)
sparse_operation_kit/sparse_operation_kit/lookup.py:418
↓ 4 callersMethodto_static
(self, indices)
sparse_operation_kit/sparse_operation_kit/dynamic_variable.py:205
↓ 4 callersFunctiontranspose
test/utest/core23_layer_test/prelu_dice_layer_test.cpp:30
↓ 4 callersFunctiontranspose
test/utest/core23_layer_test/fully_connected_layer_test.cpp:58
↓ 4 callersFunctiontrunc_half
test/utest/core23_layer_test/fm_order2_layer_test.cpp:42
↓ 4 callersMethodupdate_meta
sparse_operation_kit/kit_src/lookup/kernels/embedding_collection.cc:107
↓ 4 callersMethodupdate_per_gpu
HugeCTR/src/embeddings/embedding_collection.cpp:429
↓ 3 callersFunctionAllocateBuffers
HugeCTR/core23/buffer_factory.cpp:67
↓ 3 callersMethodDenseLayerComputeConfig
HugeCTR/src/pybind/model.cpp:66
← previousnext →201–300 of 3,072, ranked by callers