Commit cfbca2e6 authored by Peter Hawkins's avatar Peter Hawkins Committed by TensorFlower Gardener
Browse files

[XRT] Work on making XRT callable from Python via the TF eager API.

[XRT] Add an XRTReadToTensor operator that reads an XRT buffer into a host-memory Tensor. This is intended as a mirror image of XRTAllocateFromTensor, and is useful when interoperating between regular TF and XRT ops.
[XLA:Python] Add a helper for reading the ProgramShape of a computation, i.e., both argument and return value shapes. Minor refactoring to use Safe_PyObjectPtr.

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