Skip to content
This repository was archived by the owner on Jul 10, 2025. It is now read-only.

Commit f4b925c

Browse files
committed
Update user API and op definition based on comments.
1 parent 912657c commit f4b925c

File tree

1 file changed

+7
-8
lines changed

1 file changed

+7
-8
lines changed

rfcs/20200411-fuse_recv.md

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -149,8 +149,7 @@ additional devices as much as possible.
149149
Example of how to enable the FuseRecv feature:
150150

151151
```
152-
>>> config = tf.ConfigProto()
153-
>>> config.graph_options.optimizer_options.experimental.do_fuse_recv = True
152+
>>> tf.config.optimizer.set_experimental_options({"do_fuse_recv": True})
154153
```
155154

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

166165
### FuseRecv Op
167-
We introduce the FuseRecv Op and an RPC operation named FuseRecvTensorAsync in
168-
RemoteWorker and WorkerService. The FuseRecv Op definition is as follows:
166+
We introduce the _RecvV2 Op and an RPC operation named FuseRecvTensorAsync in
167+
RemoteWorker and WorkerService. The _RecvV2 Op definition is as follows:
169168

170169
```
171-
>>> REGISTER_OP("FuseRecv")
170+
>>> REGISTER_OP("_RecvV2")
172171
>>> .Output("tensor: tensor_type")
173172
>>> .Attr("tensor_type: list(type)")
174173
>>> .Attr("tensor_name: list(string)")
175-
>>> .Attr("send_device: list(string)")
176-
>>> .Attr("send_device_incarnation: list(int)")
177-
>>> .Attr("recv_device: list(string)")
174+
>>> .Attr("send_device: string")
175+
>>> .Attr("send_device_incarnation: int")
176+
>>> .Attr("recv_device: string")
178177
>>> .Attr("client_terminated: bool = false")
179178
>>> .SetIsStateful()
180179
>>> .SetShapeFn(shape_inference::UnknownShape);

0 commit comments

Comments
 (0)