Commit 07cc6474 authored by Mark Heffernan's avatar Mark Heffernan Committed by TensorFlower Gardener
Browse files

Add single-sided host send and receive operations. 

Adds a bit on kSend/kReceive instructions and their Done variants indicated whether the operations communicates with the host or another device (the default). Host send/recv operations are single-sided without a complementary recv/send instruction in another module.

Host send/recv operations are exposed in the XLA builder API as SendToHost and RecvFromHost.

PiperOrigin-RevId: 205008138
parent 1662a105
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