MCPcopy Create free account

hub / github.com/PersiaML/PERSIA / functions

Functions720 in github.com/PersiaML/PERSIA

⨍ Functions 720◇ Types & classes 217↳ Endpoints 1
↓ 1 callersMethodlog_metrics
(&self)
rust/persia-metrics/src/lib.rs:203
↓ 1 callersFunctionlookup_batched_all_slots_postprocess
( indices: &IDTypeFeatureBatch, forwarded_groups: Vec<(Vec<f32>, Vec<SignWithConfig>)>, config: &'
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:486
↓ 1 callersFunctionlookup_batched_all_slots_preprocess
( indices: &mut IDTypeFeatureBatch, config: &EmbeddingConfig, replica_size: u64, )
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:448
↓ 1 callersMethodlookup_mixed
( &self, req: (Vec<(u64, usize)>, bool), )
rust/persia-embedding-server/src/embedding_parameter_service/mod.rs:344
↓ 1 callersFunctionmemcpy
(bytes_src: &[u8], bytes_dst: &mut [u8])
rust/others/persia-common-benchmark/benches/memcpy.rs:7
↓ 1 callersFunctionndarray_f16_to_f32
(input: &Array2<f16>)
rust/persia-common/src/lib.rs:163
↓ 1 callersMethodnext
(&mut self)
rust/persia-embedding-holder/src/array_linked_list.rs:998
↓ 1 callersMethodnum_internal_shards
(&self)
rust/persia-embedding-holder/src/lib.rs:75
↓ 1 callersMethodnum_total_signs
(&self)
rust/persia-embedding-holder/src/lib.rs:67
↓ 1 callersFunctionparse_embedding_config
(config: EmbeddingConfig)
rust/persia-embedding-config/src/lib.rs:600
↓ 1 callersMethodpop_front
Removes the first element from the array and returns it, or `None` if it is empty. This operation should compute in *O*(1) time. # Examples ``` use
rust/persia-embedding-holder/src/array_linked_list.rs:649
↓ 1 callersMethodpublish_subject_ident
(&self)
rust/others/persia-nats-marcos/src/lib.rs:24
↓ 1 callersMethodpublisher_impl
(&self)
rust/others/persia-nats-marcos/src/lib.rs:190
↓ 1 callersMethodpublishing
(&self, service_type_name: String)
rust/others/persia-nats-marcos/src/lib.rs:100
↓ 1 callersMethodpush_metrics
(&self)
rust/persia-metrics/src/lib.rs:169
↓ 1 callersMethodread_from_file
(&self, file_path: String, py: Python<'a>)
rust/persia-core/src/lib.rs:409
↓ 1 callersMethodread_to_end
(&self)
rust/persia-storage/src/lib.rs:99
↓ 1 callersMethodread_to_string
(&self)
rust/persia-storage/src/lib.rs:108
↓ 1 callersMethodregister_optimizer
(&self, optimizer: &OptimizerBase)
rust/persia-core/src/lib.rs:166
↓ 1 callersFunctionregister_submodule
Register the persia core module to sys module path. Arguments: module (ModuleType): root module. root_module_path (str): root mod
persia/prelude.py:11
↓ 1 callersMethodreplica_index
(&self)
rust/persia-embedding-server/src/embedding_parameter_service/mod.rs:141
↓ 1 callersMethodreq_type
(&self)
rust/others/persia-rpc-macro/src/lib.rs:24
↓ 1 callersMethodresp_type
(&self)
rust/others/persia-nats-marcos/src/lib.rs:35
↓ 1 callersMethodresp_type
(&self)
rust/others/persia-rpc-macro/src/lib.rs:31
↓ 1 callersMethodresponder_impl
(&self)
rust/others/persia-nats-marcos/src/lib.rs:143
↓ 1 callersMethodsend_id_type_features_to_embedding_worker
( &self, batch: &mut PersiaBatch, )
rust/persia-core/src/lib.rs:342
↓ 1 callersMethodsend_non_id_type_features_to_nn_worker
(&self, batch: &PersiaBatch)
rust/persia-core/src/lib.rs:356
↓ 1 callersMethodservice_impl
(&self)
rust/others/persia-rpc-macro/src/lib.rs:177
↓ 1 callersMethodservice_method
(&self)
rust/others/persia-rpc-macro/src/lib.rs:62
↓ 1 callersFunctionsetLogger
r"""Helper function to simplify the logger setup process with provided log_level and log_filename. Also makes it possible to redirect logs abo
persia/logger.py:55
↓ 1 callersMethodset_embedding
( &self, req: Vec<HashMapEmbeddingEntry>, )
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:959
↓ 1 callersMethodset_input_channel
(&mut self, receiver: &PersiaBatchDataReceiver)
rust/persia-core/src/forward.rs:899
↓ 1 callersFunctionsetup_seed
Set the random seed for dependencies to ensure that experiments are reproducible. Arguments: seed (int): integer to use as seed for rando
persia/utils.py:13
↓ 1 callersMethodshutdown
(&self, _req: ())
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:1425
↓ 1 callersMethodshutdown_server
(&self, _req: ())
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:1404
↓ 1 callersMethodsize
(&self)
rust/persia-core/src/cuda/cuda_event_pool.rs:47
↓ 1 callersMethodspawn_backward_to_cpu_worker
(&mut self)
rust/persia-core/src/backward.rs:233
↓ 1 callersMethodspawn_backward_worker
(&mut self, num_backward_worker: usize)
rust/persia-core/src/backward.rs:304
↓ 1 callersMethodspawn_forward_worker
(&mut self, num_workers: usize)
rust/persia-core/src/forward.rs:640
↓ 1 callersMethodspawn_postprocess_worker
(&mut self)
rust/persia-core/src/forward.rs:572
↓ 1 callersMethodspawn_push
(&self)
rust/persia-metrics/src/lib.rs:188
↓ 1 callersMethodspawn_reorder_buffer_worker
(&mut self)
rust/persia-core/src/forward.rs:559
↓ 1 callersMethodsubscription
(&self, service_type_name: String)
rust/others/persia-nats-marcos/src/lib.rs:46
↓ 1 callersMethodsync_event
(&mut self)
rust/persia-core/src/cuda/mod.rs:83
↓ 1 callersFunctiontest
(model: torch.nn.Module, data_loader: DataLoader, cuda: bool)
examples/src/adult-income/train.py:43
↓ 1 callersMethodto_dldevicetype
(&self)
rust/persia-core/src/tensor.rs:196
↓ 1 callersMethodto_dldtype
Convert to [`DLDataType`].
rust/persia-core/src/tensor.rs:70
↓ 1 callersMethodto_optimizable
(self)
rust/persia-common/src/optim.rs:57
↓ 1 callersMethodtry_commit_incremental
( &self, incremental: Vec<u64>, )
rust/persia-incremental-update-manager/src/lib.rs:239
↓ 1 callersMethodupdate_all_batched_gradients
( &self, embedding_gradient_batch: &mut EmbeddingGradientBatch, indices: IDTypeFeature
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:703
↓ 1 callersMethodupdate_gradient_batched
( &self, req: (u64, EmbeddingGradientBatch), )
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:1109
↓ 1 callersMethodupdate_gradient_mixed
( &self, req: (Vec<u64>, Vec<f32>), )
rust/persia-embedding-server/src/embedding_parameter_service/mod.rs:359
↓ 1 callersMethodupdate_id_type_feature_gradient_batched
( &self, gradients: &mut GradientBatch, )
rust/persia-core/src/backward.rs:379
↓ 1 callersFunctionwait_for_compelete
( job_name: &str, namespace: &str, kubernetes_client: Client, timeout: Duration, )
k8s/src/bin/e2e.rs:153
↓ 1 callersMethodwait_for_dump_embedding
Wait for the embedding dump process.
persia/ctx.py:604
↓ 1 callersMethodwait_for_emb_dumping
(&self)
rust/persia-core/src/lib.rs:328
↓ 1 callersMethodwait_for_emb_loading
(&self)
rust/persia-core/src/lib.rs:321
↓ 1 callersMethodwait_for_load_embedding
Wait for the embedding load process.
persia/ctx.py:608
↓ 1 callersMethodwait_for_serving
(self)
persia/ctx.py:1132
↓ 1 callersMethodwaiting_for_all_embedding_server_dump
( &self, timeout_sec: usize, dst_dir: PathBuf, )
rust/persia-model-manager/src/lib.rs:200
↓ 1 callersFunctionweight_bound
(embedding: &mut [f32], weight_bound: f32)
rust/persia-simd/src/lib.rs:231
Method__del__
(self)
persia/data.py:270
Method__enter__
(self)
persia/ctx.py:253
Method__enter__
(self)
persia/helper.py:239
Method__exit__
(self, exc_type, value, trace)
persia/ctx.py:262
Method__exit__
(self, _exc_type, _exc_val, _exc_tb)
persia/helper.py:317
Method__init__
Arguments: master_port (int): master of collective communication ip address. master_addr (str, optional): master of c
persia/distributed.py:19
Method__init__
Arguments: algorithm (str): name of Bagua algorithm. options (dict): options for Bagua algorithm
persia/distributed.py:301
Method__init__
( self, *args, **kwargs, )
persia/ctx.py:320
Method__init__
Arguments: preprocess_mode (PreprocessMode): different preprocess mode effect the behavior of :meth:`.prepare_fea
persia/ctx.py:388
Method__init__
Arguments: embedding_optimizer (persia.embedding.optim.Optimizer): optimizer for the embedding parameters.
persia/ctx.py:753
Method__init__
Arguments: embedding_worker_addrs (List[str]): embedding worker address(ip:port) list.
persia/ctx.py:1115
Method__init__
(self, level: int)
persia/logger.py:15
Method__init__
( self, buffer_size: int = 10, )
persia/data.py:83
Method__init__
( self, buffer_size: int = 10, )
persia/data.py:113
Method__init__
(self)
persia/env.py:17
Method__init__
Arguments: data_loader_func (Callable, optional): data loader function that will be pickled and run on the indivi
persia/helper.py:176
Method__init__
(self)
persia/embedding/optim.py:10
Method__init__
Arguments: lr(float): learning rate. betas(tuple[float,float], optional): calculate the running averages of gradient
persia/embedding/optim.py:38
Method__init__
Arguments: lr (float): learning rate. initial_accumulator_value (float, optional): initialization accumulator value f
persia/embedding/optim.py:63
Method__init__
Arguments: emb_initialization (Tuple[float, float], optional): lower and upper bound of embedding uniform initialization.
persia/embedding/__init__.py:7
Method__init__
Arguments: name (str): name of :class:`IDTypeFeature`. data (List[np.ndarray]): A lil sparse matrix data. Requires np
persia/embedding/data.py:98
Method__init__
Arguments: name (str): name of :class:`IDTypeFeatureWithSingleID`. data (np.ndarray): :class:`IDTypeFeatureWithSingle
persia/embedding/data.py:142
Method__init__
Arguments: data (np.ndarray): Numpy array. name (str, optional): name of data.
persia/embedding/data.py:193
Method__init__
Arguments: id_type_features (List[Union[IDTypeFeatureWithSingleID, IDTypeFeature]]): categorical data whose datat
persia/embedding/data.py:329
Method__init__
( self, non_id_type_feature_data, id_type_feature_data, label_data, id
examples/src/adult-income/data_generator.py:11
Method__init__
(self, test_dir: str, batch_size: int = 128)
examples/src/adult-income/train.py:28
Method__init__
( self, dense_mlp_output_size: int = 16, sparse_mlp_output_size: int = 128 )
examples/src/adult-income/model.py:8
Method__iter__
(self)
persia/data.py:134
Method__iter__
(self)
persia/data.py:255
Method__iter__
( self, )
examples/src/adult-income/data_generator.py:34
Method__iter__
(self)
examples/src/adult-income/train.py:32
Method__len__
(self)
persia/embedding/data.py:213
Method__len__
(self)
examples/src/adult-income/data_generator.py:71
Function_data_loader
()
persia/helper.py:336
Method_detect_subprocess_terminate
(check_interval: int = 2)
persia/helper.py:296
Function_ensure_parse_env
(get_func)
persia/env.py:88
Method_enter
(self)
persia/ctx.py:411
Method_enter
(self)
persia/ctx.py:854
Method_exit
(self)
persia/ctx.py:860
← previousnext →301–400 of 720, ranked by callers