Searched refs:TcpUserCommand (Results 1 – 15 of 15) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/core/communicator/ |
| D | tcp_communicator.h | 39 const std::unordered_map<TcpUserCommand, std::string> kUserCommandToMsgType = {
40 {TcpUserCommand::kPush, "push"},
41 {TcpUserCommand::kPull, "pull"},
42 {TcpUserCommand::kCount, "count"},
43 {TcpUserCommand::kReachThreshold, "countReachThreshold"},
44 {TcpUserCommand::kResetCount, "resetCnt"},
45 {TcpUserCommand::kGetMetadata, "getMetadata"},
46 {TcpUserCommand::kUpdateMetadata, "updateMetadata"},
47 {TcpUserCommand::kCounterEvent, "counterEvent"},
48 {TcpUserCommand::kPullWeight, "pullWeight"},
[all …]
|
| D | tcp_communicator.cc | 37 TcpUserCommand user_command = static_cast<TcpUserCommand>(meta->user_cmd()); in Start()
|
| D | communicator_base.h | 36 enum class TcpUserCommand { enum
|
| /third_party/mindspore/mindspore/ccsrc/ps/core/ |
| D | instance_manager.cc | 32 message_meta->set_user_cmd(static_cast<int32_t>(TcpUserCommand::kNewInstance)); in NewInstanceAsync()
51 message_meta->set_user_cmd(static_cast<int32_t>(TcpUserCommand::kQueryInstance)); in QueryInstanceAsync()
71 message_meta->set_user_cmd(static_cast<int32_t>(TcpUserCommand::kEnableFLS)); in EnableFLSAsync()
91 message_meta->set_user_cmd(static_cast<int32_t>(TcpUserCommand::kDisableFLS)); in DisableFLSAsync()
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/ |
| D | distributed_count_service.cc | 111 …unicator_->SendPbRequest(report_count_req, counting_server_rank_, ps::core::TcpUserCommand::kCount, in Count()
151 … ps::core::TcpUserCommand::kReachThreshold, &query_cnt_enough_rsp_msg)) { in CountReachThreshold()
332 …if (!communicator_->SendPbRequest(first_count_event, i, ps::core::TcpUserCommand::kCounterEvent)) { in TriggerFirstCountEvent()
358 … if (!communicator_->SendPbRequest(last_count_event, i, ps::core::TcpUserCommand::kCounterEvent)) { in TriggerLastCountEvent()
|
| D | iteration.cc | 349 …if (!communicator_->SendPbRequest(sync_iter_req, kLeaderServerRank, ps::core::TcpUserCommand::kSyn… in SyncIteration()
401 ps::core::TcpUserCommand::kNotifyLeaderToNextIter)) { in NotifyLeaderMoveToNextIteration()
456 …if (!communicator_->SendPbRequest(prepare_next_iter_req, i, ps::core::TcpUserCommand::kPrepareForN… in BroadcastPrepareForNextIterRequest()
467 …!communicator_->SendPbRequest(prepare_next_iter_req, rank, ps::core::TcpUserCommand::kPrepareForNe… in BroadcastPrepareForNextIterRequest()
511 …if (!communicator_->SendPbRequest(proceed_to_next_iter_req, i, ps::core::TcpUserCommand::kProceedT… in BroadcastMoveToNextIterRequest()
576 … if (!communicator_->SendPbRequest(end_last_iter_req, i, ps::core::TcpUserCommand::kEndLastIter)) { in BroadcastEndLastIterRequest()
|
| D | distributed_metadata_store.cc | 103 …if (!communicator_->SendPbRequest(metadata_with_name, stored_rank, ps::core::TcpUserCommand::kUpda… in UpdateMetadata()
140 …if (!communicator_->SendPbRequest(get_metadata_req, stored_rank, ps::core::TcpUserCommand::kGetMet… in GetMetadata()
|
| /third_party/mindspore/mindspore/ccsrc/fl/worker/ |
| D | fl_worker.h | 70 …bool SendToServer(uint32_t server_rank, const void *data, size_t size, ps::core::TcpUserCommand co…
|
| D | fl_worker.cc | 98 …SendToServer(uint32_t server_rank, const void *data, size_t size, ps::core::TcpUserCommand command, in SendToServer()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/ |
| D | push_metrics_kernel.h | 65 … fbb_->GetSize(), ps::core::TcpUserCommand::kPushMetrics, in Launch()
|
| D | start_fl_job_kernel.h | 43 … fbb_->GetSize(), ps::core::TcpUserCommand::kStartFLJob, in Launch()
|
| D | get_model_kernel.h | 49 … fbb_->GetSize(), ps::core::TcpUserCommand::kGetModel, in Launch()
|
| D | update_model_kernel.h | 55 … fbb_->GetSize(), ps::core::TcpUserCommand::kUpdateModel, in Launch()
|
| D | fused_push_weight_kernel.h | 80 ps::core::TcpUserCommand::kPushWeight, in Launch()
|
| D | fused_pull_weight_kernel.h | 80 …0, fbb->GetBufferPointer(), fbb->GetSize(), ps::core::TcpUserCommand::kPullWeight, &pull_weight_rs… in Launch()
|