| 167 | } |
| 168 | |
| 169 | void SetSendRecvAttrs(const PartitionOptions& opts, const Edge* edge, |
| 170 | NodeDefBuilder* builder) { |
| 171 | builder->Attr("tensor_name", |
| 172 | strings::StrCat("edge_", edge->id(), "_", edge->src()->name())); |
| 173 | builder->Attr("send_device", edge->src()->assigned_device_name()); |
| 174 | builder->Attr("send_device_incarnation", |
| 175 | static_cast<int64>( |
| 176 | opts.get_incarnation(edge->src()->assigned_device_name()))); |
| 177 | builder->Attr("recv_device", edge->dst()->assigned_device_name()); |
| 178 | builder->Attr("client_terminated", false); |
| 179 | } |
| 180 | |
| 181 | void SetFuseRecvAttrs(const PartitionOptions& opts, |
| 182 | const std::vector<const Edge*>& edges, |