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

Internal Variant API allowing registering Variants to be copied from/to GPU. 

Adds a test in the variant_op_copy_test.

Modifies the base GPUDevice to use this registry if it sees a singleton variant.

Modifies the rendezvous manager to do the same.

PiperOrigin-RevId: 170908757
parent 8fb14b14
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