Commit 4ea51ae9 authored by Andiry Xu's avatar Andiry Xu Committed by TensorFlower Gardener
Browse files

Change _Send/_Recv attrs in VirtualScheduler 

Change src_device_ to send_device and dst_device_ to recv_device. This complies
with tensorflow naming, so that VirtualScheduler can handle graphs generated
on inspectz with _Send/_Recv nodes and AutoGrappler does not need to remove
them.

PiperOrigin-RevId: 221660625
parent dcf390ed
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment