Commit c5328715 authored by Skye Wanderman-Milne's avatar Skye Wanderman-Milne Committed by TensorFlower Gardener
Browse files

Register resource GPU kernels for NextIteration and Exit.

Prior to this change, accessing a GPU resource variable from a while
loop would fail, because the resource NextIteration and Exit nodes
would cause the placer to put the whole loop body on CPU, including
variable reads, etc. which must be colocated with the resource.

PiperOrigin-RevId: 226210621
parent d2e68e72
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment