Commit eea28cb2 authored by Eugene Brevdo's avatar Eugene Brevdo Committed by TensorFlower Gardener
Browse files

[TF Variants] The ResourceAssignOp now supports Variants on GPU. 

This allows Variants to sit on resource variables; before, though the
ReadValue op was enabled for Variants on GPU, because assignment happened
on CPU, Variant-based Resource Variables always had to reside on CPU
due to the associated colocation constraints.

PiperOrigin-RevId: 178996911
parent ed0e2508
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