| 33 | namespace graph { |
| 34 | |
| 35 | Node* Send(Graph* g, Node* input, const string& tensor, const string& sender, |
| 36 | const uint64 sender_incarnation, const string& receiver) { |
| 37 | Node* ret; |
| 38 | TF_CHECK_OK(NodeBuilder(g->NewName("n"), "_Send") |
| 39 | .Input(input, 0) |
| 40 | .Attr("tensor_name", tensor) |
| 41 | .Attr("send_device", sender) |
| 42 | .Attr("send_device_incarnation", |
| 43 | static_cast<int64>(sender_incarnation)) |
| 44 | .Attr("recv_device", receiver) |
| 45 | .Finalize(g, &ret)); |
| 46 | return ret; |
| 47 | } |
| 48 | |
| 49 | Node* Recv(Graph* g, const string& tensor, const string& type, |
| 50 | const string& sender, const uint64 sender_incarnation, |
no test coverage detected