| 126 | } |
| 127 | |
| 128 | Node* FileSliceRecv(Graph* g, const string& tensor, const string& sender, |
| 129 | const uint64 sender_incarnation, const string& receiver, |
| 130 | const string& recv_dir, const int32 slice_size, |
| 131 | const int64 timeout_ms) { |
| 132 | Node* ret; |
| 133 | TF_CHECK_OK(NodeBuilder(g->NewName("FileSliceRecv"), "_FileSliceRecv") |
| 134 | .Attr("tensor_name", tensor) |
| 135 | .Attr("send_device", sender) |
| 136 | .Attr("send_device_incarnation", |
| 137 | static_cast<int64>(sender_incarnation)) |
| 138 | .Attr("recv_device", receiver) |
| 139 | .Attr("recv_dir", recv_dir) |
| 140 | .Attr("slice_size", slice_size) |
| 141 | .Attr("timeout_ms", timeout_ms) |
| 142 | .Finalize(g, &ret)); |
| 143 | |
| 144 | return ret; |
| 145 | } |
| 146 | |
| 147 | Node* SliceSend(Graph* g, Node* input, const string& tensor, |
| 148 | const string& sender, const uint64 sender_incarnation, |
no test coverage detected