Qiao Zhang c98ac56bc2 Refactor PJRT. ... - Add TransferToInfeed and TransferFromOutfeed to PjRtDevice's methods. PiperOrigin-RevId: 338105874 Change-Id: I629b2efa27394bc99b26371c3de779b1104eea4f		2020-10-20 12:08:59 -07:00
..
distributed	Remove usage of tf_proto_library_cc rule.	2020-09-25 12:38:45 -07:00
BUILD	[XLA:Python] Add TPU backend.	2020-10-14 16:07:01 -07:00
cpu_device.cc	[PJRT] Change client factory methods to return a std::unique_ptr<> instead of a std::shared_ptr<>.	2020-10-08 07:30:47 -07:00
cpu_device.h	[PJRT] Change client factory methods to return a std::unique_ptr<> instead of a std::shared_ptr<>.	2020-10-08 07:30:47 -07:00
event_pool.cc
event_pool.h
gpu_multistream_test.cc	Refactor PJRT.	2020-10-20 11:57:11 -07:00
interpreter_device.cc	[PJRT] Change client factory methods to return a std::unique_ptr<> instead of a std::shared_ptr<>.	2020-10-08 07:30:47 -07:00
interpreter_device.h	[PJRT] Change client factory methods to return a std::unique_ptr<> instead of a std::shared_ptr<>.	2020-10-08 07:30:47 -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] Change client factory methods to return a std::unique_ptr<> instead of a std::shared_ptr<>.	2020-10-08 07:30:47 -07:00
nvidia_gpu_device.h	[PJRT] Change client factory methods to return a std::unique_ptr<> instead of a std::shared_ptr<>.	2020-10-08 07:30:47 -07:00
pjrt_client.cc	Refactor PJRT.	2020-10-20 12:08:59 -07:00
pjrt_client.h	Refactor PJRT.	2020-10-20 12:08:59 -07:00
semaphore_test.cc
semaphore.cc
semaphore.h
tpu_client.cc	[XLA:Python] Add TPU backend.	2020-10-14 16:07:01 -07:00
tpu_client.h	[XLA:PJRT] Add TPU client and associated libtpu.so initializer.	2020-10-06 15:15:22 -07:00
tracked_device_buffer_test.cc
tracked_device_buffer.cc
tracked_device_buffer.h
worker_thread.cc
worker_thread.h