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

Split non-Python PJRT classes into their own directory.

2020-05-01 10:01:09 -07:00

worker_thread.cc

…

worker_thread.h

…