Skip to content

Commit

Permalink
Update user API and op definition based on comments.
Browse files Browse the repository at this point in the history
  • Loading branch information
liutongxuan committed May 12, 2020
1 parent 912657c commit f4b925c
Showing 1 changed file with 7 additions and 8 deletions.
15 changes: 7 additions & 8 deletions rfcs/20200411-fuse_recv.md
Expand Up @@ -149,8 +149,7 @@ additional devices as much as possible.
Example of how to enable the FuseRecv feature:

```
>>> config = tf.ConfigProto()
>>> config.graph_options.optimizer_options.experimental.do_fuse_recv = True
>>> tf.config.optimizer.set_experimental_options({"do_fuse_recv": True})
```

### Compatibility
Expand All @@ -164,17 +163,17 @@ Example of how to enable the FuseRecv feature:
## Detailed Design

### FuseRecv Op
We introduce the FuseRecv Op and an RPC operation named FuseRecvTensorAsync in
RemoteWorker and WorkerService. The FuseRecv Op definition is as follows:
We introduce the _RecvV2 Op and an RPC operation named FuseRecvTensorAsync in
RemoteWorker and WorkerService. The _RecvV2 Op definition is as follows:

```
>>> REGISTER_OP("FuseRecv")
>>> REGISTER_OP("_RecvV2")
>>> .Output("tensor: tensor_type")
>>> .Attr("tensor_type: list(type)")
>>> .Attr("tensor_name: list(string)")
>>> .Attr("send_device: list(string)")
>>> .Attr("send_device_incarnation: list(int)")
>>> .Attr("recv_device: list(string)")
>>> .Attr("send_device: string")
>>> .Attr("send_device_incarnation: int")
>>> .Attr("recv_device: string")
>>> .Attr("client_terminated: bool = false")
>>> .SetIsStateful()
>>> .SetShapeFn(shape_inference::UnknownShape);
Expand Down

0 comments on commit f4b925c

Please sign in to comment.