STT-tensorflow/tensorflow/compiler/xla/pjrt
Tamara Norman 0868ca7bb2 Allow a shape to be passed to CopyToHostAsync
PiperOrigin-RevId: 317611333
Change-Id: I4526f9dbd1b223eb23fe928326afca0eb133c2f5
2020-06-22 01:55:19 -07:00
..
distributed
BUILD Allow a shape to be passed to CopyToHostAsync 2020-06-22 01:55:19 -07:00
cpu_device.cc [PJRT] Fix potential misuse of PjRtBuffer::FromHostBuffer. 2020-06-16 06:59:42 -07:00
cpu_device.h
event_pool.cc
event_pool.h
gpu_multistream_test.cc [PJRT] Fix potential misuse of PjRtBuffer::FromHostBuffer. 2020-06-16 06:59:42 -07:00
interpreter_device.cc [PJRT] Fix potential misuse of PjRtBuffer::FromHostBuffer. 2020-06-16 06:59:42 -07:00
interpreter_device.h [XLA:Python] Plumb in the HLO interpreter platform. 2020-06-02 17:40:19 -07:00
local_device_state.cc [pjrt] Refresh stream error status in strategic places to flush out silent failures. 2020-06-18 16:28:27 -07:00
local_device_state.h
nvidia_gpu_device.cc [PJRT] Fix potential misuse of PjRtBuffer::FromHostBuffer. 2020-06-16 06:59:42 -07:00
nvidia_gpu_device.h
pjrt_client.cc Allow a shape to be passed to CopyToHostAsync 2020-06-22 01:55:19 -07:00
pjrt_client.h Allow a shape to be passed to CopyToHostAsync 2020-06-22 01:55:19 -07:00
semaphore_test.cc
semaphore.cc
semaphore.h
tracked_device_buffer_test.cc
tracked_device_buffer.cc
tracked_device_buffer.h
worker_thread.cc
worker_thread.h