Home
last modified time | relevance | path

Searched refs:scale_in_message (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/ps/core/
Dleader_scaler.cc49 ScaleInMessage scale_in_message; in ScaleInAsync() local
50 scale_in_message.set_worker_num(manager.worker_num()); in ScaleInAsync()
51 scale_in_message.set_server_num(manager.server_num()); in ScaleInAsync()
52 scale_in_message.set_is_node_scale_in(is_node_scale_in); in ScaleInAsync()
54 …if (!node_->SendMessageSync(client, message_meta, Protos::PROTOBUF, scale_in_message.SerializeAsSt… in ScaleInAsync()
55 scale_in_message.ByteSizeLong())) { in ScaleInAsync()
Dabstract_node.cc764 ScaleInMessage scale_in_message; in ProcessScaleIn() local
765 CHECK_RETURN_TYPE(scale_in_message.ParseFromArray(data, SizeToInt(size))); in ProcessScaleIn()
766 int32_t worker_num = scale_in_message.worker_num(); in ProcessScaleIn()
767 int32_t server_num = scale_in_message.server_num(); in ProcessScaleIn()
770 is_current_node_scale_in_ = scale_in_message.is_node_scale_in(); in ProcessScaleIn()