| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/ |
| D | _ps_context.py | 179 def _insert_hash_table_size(name, cache_vocab_size, embedding_size, vocab_size, param_key=-1): argument
237 def _store_warm_up_ptr_by_tensor(param_key, tensor): argument
241 def _store_warm_up_ptr_by_tensor_list(param_key, key_tensor, value_tensor, status_tensor): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/distributed/embedding_cache/ |
| D | embedding_cache_utils.cc | 72 … size_t embedding_size, size_t vocab_size, int32_t param_key) { in InsertHashTableSize()
249 int32_t param_key = item.second.param_key_; in DumpHashTables() local
263 int32_t EmbeddingCacheTableManager::StoreWarmUpPtr(const int32_t param_key, const tensor::TensorPtr… in StoreWarmUpPtr()
267 int32_t EmbeddingCacheTableManager::StoreWarmUpPtr(const int32_t param_key, const tensor::TensorPtr… in StoreWarmUpPtr()
283 const HashTableInfo *EmbeddingCacheTableManager::FindHashTablesByParamKey(const int param_key) { in FindHashTablesByParamKey()
541 void EmbeddingStorageManager::Add(int32_t param_key, in Add()
547 std::shared_ptr<storage::AbstractEmbeddingStorage> EmbeddingStorageManager::Get(int32_t param_key) { in Get()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/ps/ |
| D | ps_context.cc | 356 int32_t PSContext::StoreWarmUpPtrByTensor(const int32_t param_key, const tensor::TensorPtr &tensor_… in StoreWarmUpPtrByTensor()
365 int32_t PSContext::StoreWarmUpPtrByTensorList(const int32_t param_key, const tensor::TensorPtr &key… in StoreWarmUpPtrByTensorList()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/distributed/embedding_cache/ |
| D | embedding_cache_test.cc | 51 int32_t param_key = 0; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/ |
| D | embedding_cache_prefetch_actor.cc | 43 … uint32_t dst_rank, const std::string &cache_operation, int32_t param_key) { in GenerateInterProcessEdge()
60 … int32_t param_key, device::DeviceContext *cpu_device_context) { in CreateSenderReceiverPair()
85 int32_t GetCacheOpsServiceId(const std::string &cache_operation, int32_t param_key) { in GetCacheOpsServiceId()
940 bool EmbeddingCachePrefetchActor::PullEembeddingsFromRemote(int32_t param_key, const int *ids, size… in PullEembeddingsFromRemote()
993 bool EmbeddingCachePrefetchActor::DoPushEmbeddingsToRemote(int32_t param_key, const int *ids, size_… in DoPushEmbeddingsToRemote()
1027 bool EmbeddingCachePrefetchActor::PushEmbeddingsToRemote(int32_t param_key, const int *ids, size_t … in PushEmbeddingsToRemote()
1151 …ol EmbeddingCachePrefetchActor::SendToRemote(const std::string &cache_operation, int32_t param_key, in SendToRemote()
1199 int32_t param_key, in ReceiveFromRemote()
|
| /third_party/openssl/test/ |
| D | acvp_test.c | 340 EVP_PKEY *param_key = NULL; in dsa_paramgen() local
354 EVP_PKEY *param_key = NULL, *key = NULL; in dsa_keygen() local
373 EVP_PKEY *param_key = NULL, *key = NULL; in dsa_keygen_test() local
413 EVP_PKEY *param_key = NULL; in dsa_paramgen_test() local
525 EVP_PKEY *param_key = NULL; in dsa_pqver_test() local
|
| D | dsatest.c | 152 EVP_PKEY *param_key = NULL, *key = NULL; in dsa_keygen_test() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/ |
| D | map_tensor_py.cc | 97 const MapTensorPtr &map_tensor, int32_t param_key, bool incremental) { in ExportPersistentSliceAsNumpy()
|
| D | tensor_py.cc | 258 void FillSliceData(const int32_t param_key, const int slice_index) { in FillSliceData()
662 py::array TensorPy::AsNumpyOfSlice(const Tensor &tensor, const int32_t param_key, const int slice_i… in AsNumpyOfSlice()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/cache_embedding/ |
| D | ps_embedding_cache_inserter.cc | 191 void PsEmbeddingCacheInserter::SetSendNodeAttr(const CNodePtr &send_node, int32_t param_key, in SetSendNodeAttr()
232 int32_t param_key = item.first; in SetRecvNodeAttr() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/backend/distributed/embedding_cache/ |
| D | embedding_cache_utils.h | 393 …bool Exists(int32_t param_key) const { return embedding_storages_.find(param_key) != embedding_sto… in Exists()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/layer/ |
| D | embedding.py | 477 def _set_cache_enable_and_key_for_pserver(self, param_key): argument
|
| /third_party/openssl/crypto/evp/ |
| D | ctrl_params_translate.c | 262 const char *param_key; /* The corresponding OSSL_PARAM key */ member
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/common/ |
| D | tensor.py | 1074 def asnumpy_of_slice_persistent_data(self, param_key, slice_index): argument
|