First checkin of XRT primitives to call XLA computations directly via
the TensorFlow infrastructure. These primitives will allow direct access to XLA via Cloud TPU, and also provide an easy path to integrate custom XLA computations with other (distributed) TensorFlow computations on CPU and GPU. The API is *very experimental* and subject to change, perhaps substantially in the short term. PiperOrigin-RevId: 211006371
Loading
Please sign in to comment