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
↓ 998 callersMethodadd
HugeCTR/src/pybind/model_compile.cpp:85
↓ 965 callersMethodappend
HugeCTR/include/data_generator.hpp:177
↓ 868 callersMethodDenseLayer
HugeCTR/src/pybind/model.cpp:198
↓ 824 callersMethodsize
HugeCTR/core23/tensor.cpp:53
↓ 700 callersMethoddata
test/utest/core23/basic_buffer_test.cpp:45
↓ 559 callersMethodpush_back
* Push back a value to this object. * @param value the value to be pushed back. */
HugeCTR/include/data_readers/csr23.hpp:101
↓ 460 callersMethodget
test/utest/embedding/cpu_hashtable.hpp:45
↓ 369 callersMethodshape
HugeCTR/core23/tensor_params.hpp:49
↓ 287 callersMethoddata_type
HugeCTR/core23/tensor.cpp:57
↓ 280 callersMethoddevice
HugeCTR/core23/tensor.cpp:56
↓ 273 callersMethodend
HugeCTR/include/data_readers/data_reader_worker_group.hpp:152
↓ 248 callersMethodbegin
HugeCTR/include/data_readers/multi_hot/detail/batch_locations.hpp:118
↓ 232 callersMethoddata
HugeCTR/core23/tensor.cpp:48
↓ 198 callersFunctioncopy_sync
HugeCTR/core23/tensor_operations.cpp:58
↓ 193 callersMethodsize
(self)
sparse_operation_kit/sparse_operation_kit/dynamic_variable.py:234
↓ 166 callersMethodvalue
test/utest/core23_layer_test/elu_layer_test.cpp:37
↓ 162 callersMethodget_device_id
HugeCTR/include/loss.hpp:115
↓ 159 callersMethodrank
(self)
sparse_operation_kit/sparse_operation_kit/communication.py:126
↓ 154 callersMethodresize
HugeCTR/core23/details/low_level_cuda_allocator.hpp:39
↓ 144 callersMethodget_local_gpu
HugeCTR/core/hctr_impl/hctr_backend.hpp:62
↓ 143 callersMethodreshape
HugeCTR/core23/tensor.cpp:34
↓ 139 callersMethodnum_bytes
HugeCTR/core23/tensor.cpp:55
↓ 135 callersMethodinsert
test/utest/embedding/cpu_hashtable.hpp:28
↓ 133 callersMethodsize
HugeCTR/embedding_storage/dynamic_embedding_cpu.hpp:413
↓ 123 callersFunctionconvert
(src_csv, src_config, dst, batch_size, segmentation)
tools/criteo_predict/criteo2predict.py:38
↓ 123 callersMethodnum_elements
HugeCTR/core23/tensor.cpp:54
↓ 119 callersMethodfind
HugeCTR/include/embedding_training_cache/hmem_cache/sparse_model_file_ts.hpp:77
↓ 117 callersMethodInput
HugeCTR/src/pybind/model.cpp:118
↓ 110 callersMethodapply_gradients
(self, grads_and_vars, global_step=None, name=None)
sparse_operation_kit/sparse_operation_kit/optimizer.py:383
↓ 106 callersMethodallreduce
(self, tensor, op)
sparse_operation_kit/sparse_operation_kit/communication.py:150
↓ 99 callersMethodsize
* Get the total number of devices. */
HugeCTR/include/device_map.hpp:160
↓ 92 callersMethodinit
HugeCTR/src/pipeline.cpp:76
↓ 90 callersMethodget_stream
HugeCTR/core/hctr_impl/hctr_backend.hpp:43
↓ 89 callersMethodrun
(self, func, *args, **kwargs)
sparse_operation_kit/documents/tutorials/DLRM/utils.py:141
↓ 81 callersMethodflatten
HugeCTR/core23/tensor_container.hpp:85
↓ 81 callersMethodget_local_gpu_count
HugeCTR/core/hctr_impl/hctr_backend.hpp:72
↓ 80 callersMethodDataReaderSparseParam
HugeCTR/include/common.hpp:245
↓ 76 callersMethodempty
HugeCTR/core23/tensor.cpp:61
↓ 74 callersMethodSparseEmbedding
HugeCTR/src/pybind/model.cpp:161
↓ 73 callersMethodwrite
HugeCTR/src/io/s3_filesystem.cpp:212
↓ 72 callersFunctionget_default_gpu
test/utest/test_utils.hpp:253
↓ 71 callersMethodbroadcast
(self, tensor, root)
sparse_operation_kit/sparse_operation_kit/communication.py:165
↓ 69 callersMethodembedding_lookup
HugeCTR/include/embeddings/embedding_collection.hpp:136
↓ 69 callersMethodget
HugeCTR/embedding_storage/weight_io/data_info.hpp:73
↓ 60 callersMethodread
HugeCTR/src/io/s3_filesystem.cpp:233
↓ 57 callersMethodload
HugeCTR/src/io/file_loader.cpp:62
↓ 52 callersMethodbuffer_params
HugeCTR/core23/tensor_params.hpp:77
↓ 52 callersMethoddims
HugeCTR/core23/tensor.cpp:50
↓ 51 callersMethodclear
HugeCTR/embedding_storage/dynamic_embedding_cpu.hpp:443
↓ 49 callersMethodreset
HugeCTR/src/collectives/ib_proxy.cpp:56
↓ 46 callersMethodlocal_rank
(self)
sparse_operation_kit/sparse_operation_kit/communication.py:132
↓ 45 callersMethodfprop
test/utest/core23_layer_test/trainable_layer_test.cpp:62
↓ 44 callersMethodcast
HugeCTR/include/pybind/training_callback.hpp:60
↓ 43 callersMethodModel
HugeCTR/src/pybind/model.cpp:316
↓ 43 callersMethodfit
HugeCTR/src/pybind/model.cpp:691
↓ 42 callersFunctionalign_to_even
test/utest/test_utils.hpp:189
↓ 42 callersMethodfill
test/utest/test_utils.hpp:232
↓ 41 callersMethodbprop
test/utest/core23_layer_test/trainable_layer_test.cpp:67
↓ 40 callersMethodDataReaderParams
HugeCTR/src/pybind/model.cpp:71
↓ 40 callersFunctionGetRandomBufferChannel
HugeCTR/core23/buffer_channel_helpers.cpp:43
↓ 40 callersMethodcompile
HugeCTR/src/pybind/model_compile.cpp:715
↓ 40 callersMethodinitialize
HugeCTR/src/core23_network.cpp:235
↓ 40 callersFunctionnormal_sync_cpu
test/utest/test_utils.hpp:33
↓ 38 callersMethodat
HugeCTR/include/data_readers/multi_hot/detail/batch_locations.hpp:125
↓ 38 callersMethodbuffer_channel
HugeCTR/core23/tensor_params.hpp:83
↓ 38 callersMethodreserve
HugeCTR/include/general_buffer2.hpp:219
↓ 36 callersMethodassign
(self, *args, **kwargs)
sparse_operation_kit/sparse_operation_kit/dynamic_variable.py:399
↓ 36 callersMethodtype
HugeCTR/core23/device.hpp:42
↓ 35 callersMethodget_global_gpu_count
HugeCTR/core/hctr_impl/hctr_backend.hpp:74
↓ 34 callersFunctionabs
test/utest/test_utils.hpp:55
↓ 33 callersMethodinit
HugeCTR/embedding/common.cpp:403
↓ 30 callersMethodat
HugeCTR/core23/tensor_container.hpp:137
↓ 30 callersMethodget_ptr
HugeCTR/include/metrics.hpp:151
↓ 29 callersFunctionfind_item_in_map
HugeCTR/include/utils.hpp:323
↓ 29 callersMethodget_file_size
HugeCTR/src/io/s3_filesystem.cpp:165
↓ 27 callersMethodstr
HugeCTR/core23/shape.cpp:36
↓ 26 callersMethodget_device_id
HugeCTR/include/data_readers/multi_hot/detail/device_transfer.hpp:30
↓ 26 callersFunctionhas_key_
HugeCTR/include/parser.hpp:244
↓ 26 callersMethodis_open
HugeCTR/src/data_readers/file_source_parquet.cpp:140
↓ 26 callersMethodsummary
HugeCTR/src/pybind/model_compile.cpp:769
↓ 25 callersFunctioncopy_async
HugeCTR/include/tensor2.hpp:333
↓ 25 callersMethodindex
HugeCTR/core23/device.hpp:43
↓ 25 callersFunctionsok_tsl_status
sparse_operation_kit/kit_src/common/utils_experiment.h:26
↓ 24 callersMethodcompute
( self, mp_ev, model_offset, num_key_in_bucket_for_combiner, model_com
notebooks/prototype_embedding_collection/op.py:443
↓ 24 callersMethoddump
test/utest/embedding/cpu_hashtable.hpp:58
↓ 24 callersMethodget_value_ptr
HugeCTR/include/tensor2.hpp:298
↓ 24 callersMethodis_static
(self)
sparse_operation_kit/sparse_operation_kit/dynamic_variable.py:202
↓ 22 callersMethodget_batch_size_per_gpu
HugeCTR/include/embeddings/embedding_data.hpp:55
↓ 21 callersMethodget
(self, flatten_ids, length)
sparse_operation_kit/documents/tutorials/DenseDemo/models.py:113
↓ 21 callersMethodget_global_gpu_id
HugeCTR/core/hctr_impl/hctr_backend.hpp:68
↓ 21 callersMethodget_num
HugeCTR/include/data_generator.hpp:101
↓ 21 callersMethodget_ptr
HugeCTR/include/tensor2.hpp:189
↓ 21 callersMethodget_rowoffset_ptr
HugeCTR/include/tensor2.hpp:304
↓ 21 callersMethodget_slot
(self, var, name)
sparse_operation_kit/sparse_operation_kit/optimizer.py:155
↓ 20 callersMethodconvert_to_int
(cls, tf_data_type)
sparse_operation_kit/sparse_operation_kit/dump_load.py:81
↓ 20 callersFunctioncopy
HugeCTR/include/tensor2.hpp:358
↓ 19 callersMethodeval
HugeCTR/src/pybind/model.cpp:1140
↓ 19 callersMethodget_weights
HugeCTR/include/trainable_layer.hpp:157
↓ 19 callersMethodnnz
HugeCTR/include/tensor2.hpp:318
↓ 19 callersMethodsummary
(self)
sparse_operation_kit/example/sok_train_hps_inference/sok_train.py:229
next →1–100 of 3,072, ranked by callers