| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | optimizer_info_builder.h | 33 explicit OptimizerInfoBuilder(size_t worker_num) : worker_num_(worker_num) {} in OptimizerInfoBuilder()
45 virtual void BuildOutputs(OptimizerInfo *info, size_t worker_num) {} in BuildOutputs()
57 explicit MomentumOptimInfoBuilder(size_t worker_num) : OptimizerInfoBuilder(worker_num) {} in MomentumOptimInfoBuilder()
66 explicit SparseAdamOptimInfoBuilder(size_t worker_num) : OptimizerInfoBuilder(worker_num) {} in SparseAdamOptimInfoBuilder()
75 explicit SparseFtrlOptimInfoBuilder(size_t worker_num) : OptimizerInfoBuilder(worker_num) {} in SparseFtrlOptimInfoBuilder()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/ |
| D | apply_momentum_ps_kernel.h | 30 ApplyMomentumPSKernel(size_t rank_id, size_t pserver_num, size_t worker_num) in ApplyMomentumPSKernel()
|
| D | sparse_apply_adam_ps_kernel.h | 31 SparseApplyAdamPSKernel(size_t rank_id, size_t pserver_num, size_t worker_num) in SparseApplyAdamPSKernel()
|
| D | embedding_look_up_ps_kernel.h | 30 EmbeddingLookUpPSKernel(size_t rank_id, size_t pserver_num, size_t worker_num) in EmbeddingLookUpPSKernel()
|
| D | sparse_apply_lazy_adam_ps_kernel.h | 30 SparseApplyLazyAdamPSKernel(size_t rank_id, size_t pserver_num, size_t worker_num) in SparseApplyLazyAdamPSKernel()
|
| D | sparse_apply_ftrl_ps_kernel.h | 31 SparseApplyFtrlPSKernel(size_t rank_id, size_t pserver_num, size_t worker_num) in SparseApplyFtrlPSKernel()
|
| D | pserver_kernel.h | 30 PServerKernel(size_t rank_id, size_t pserver_num, size_t worker_num) in PServerKernel()
|
| /third_party/mindspore/mindspore/ccsrc/ps/core/ |
| D | cluster_metadata.h | 36 uint32_t worker_num; member
|
| D | node_recovery.cc | 29 …int32_t worker_num = std::strtol(recovery_storage_->Get(kRecoveryWorkerNum, "").c_str(), nullptr, … in Recover() local
|
| /third_party/mindspore/tests/st/fl/cross_silo_lenet/ |
| D | run_cross_silo_lenet_sched.py | 32 worker_num = args.worker_num variable
|
| D | run_cross_silo_lenet_worker.py | 37 worker_num = args.worker_num variable
|
| D | run_cross_silo_lenet_server.py | 43 worker_num = args.worker_num variable
|
| /third_party/mindspore/mindspore/ccsrc/ps/core/protos/ |
| D | comm.proto | 127 int32 worker_num = 2; field
149 int32 worker_num = 1; field
157 int32 worker_num = 1; field
|
| /third_party/mindspore/tests/st/fl/cross_silo_femnist/ |
| D | run_cross_silo_femnist_sched.py | 33 worker_num = args.worker_num variable
|
| D | run_cross_silo_femnist_worker.py | 38 worker_num = args.worker_num variable
|
| D | run_cross_silo_femnist_server.py | 44 worker_num = args.worker_num variable
|
| /third_party/mindspore/tests/ut/cpp/ps/core/ |
| D | cluster_metadata_test.cc | 37 std::string worker_num = "1"; in TEST_F() local
|
| D | cluster_available_timeout_test.cc | 35 std::string worker_num = "1"; in TEST_F() local
|
| /third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/ |
| D | run_cross_silo_fasterrcnn_sched.py | 33 worker_num = args.worker_num variable
|
| D | run_cross_silo_fasterrcnn_worker.py | 36 worker_num = args.worker_num variable
|
| D | run_cross_silo_fasterrcnn_server.py | 45 worker_num = args.worker_num variable
|
| /third_party/mindspore/tests/st/fl/albert/ |
| D | run_hybrid_train_sched.py | 36 worker_num = args.worker_num variable
|
| /third_party/mindspore/tests/st/fl/hybrid_lenet/ |
| D | run_hybrid_train_sched.py | 37 worker_num = args.worker_num variable
|
| D | run_hybrid_train_worker.py | 39 worker_num = args.worker_num variable
|
| /third_party/mindspore/tests/st/fl/mobile/ |
| D | run_mobile_sched.py | 39 worker_num = args.worker_num variable
|