Home
last modified time | relevance | path

Searched refs:set_node_id (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/ps/core/
Dabstract_node.cc33 register_message.set_node_id(node_info_.node_id_); in Register()
116 scale_out_done_message.set_node_id(node_info_.node_id_); in set_scale_out_done()
136 scale_in_done_message.set_node_id(node_info_.node_id_); in set_scale_in_done()
156 event_message.set_node_id(node_info_.node_id_); in BroadcastEvent()
541 heartbeat_message.set_node_id(node_info_.node_id_); in Heartbeat()
614 fetch_servers.set_node_id(node_info_.node_id_); in FetchServers()
Dnode_manager.cc159 servers_meta.set_node_id(it->second.node_id_); in FetchAllNodesMeta()
Dscheduler_node.cc169 register_resp_message.set_node_id(node_id); in ProcessRegister()
/third_party/mindspore/mindspore/ccsrc/profiler/device/ascend/
Dmemory_profiling.cc88 node_mem->set_node_id(node.GetNodeId()); in MemoryToPB()
/third_party/mindspore/mindspore/ccsrc/ps/
Dps_context.h185 void set_node_id(const std::string &node_id);
Dps_context.cc425 void PSContext::set_node_id(const std::string &node_id) { node_id_ = node_id; } in set_node_id() function in mindspore::ps::PSContext