This repository was archived by the owner on Jul 10, 2025. It is now read-only.
File tree Expand file tree Collapse file tree 1 file changed +7
-8
lines changed Expand file tree Collapse file tree 1 file changed +7
-8
lines changed Original file line number Diff line number Diff line change @@ -149,8 +149,7 @@ additional devices as much as possible.
149149Example 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);
You can’t perform that action at this time.
0 commit comments