Commit b027ac97 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

This CL fixes a bug preventing Eager tapes from working remotely. 

Previously, tapes would attempt to access remote TensorHandle tensors directly (unsupported remotely) to get their shapes, causing an error.
They now access the remote shape via a new TensorHandle `Shape` method, which can unify local and remote TensorHandle shape accesses.

This CL also adds some tests to ensure taping during remote execution works.

PiperOrigin-RevId: 204819435
parent dee95616
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