experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/compiler/xla/pjrt
History
Qiao Zhang ee541f2cf6 Internal changes only. 
PiperOrigin-RevId: 352121854
Change-Id: I9e7679109d15dd4a8ec75cfc295c18dd3c1eedff
2021-01-15 18:51:23 -08:00
..
distributed
Internal change
2021-01-14 16:56:45 -08:00
BUILD
Internal changes only.
2021-01-15 18:51:23 -08:00
cpu_device.cc
Split StreamExecutor impl of pjrt_client into its own files.
2020-12-10 10:07:54 -08:00
cpu_device.h
Split StreamExecutor impl of pjrt_client into its own files.
2020-12-10 10:07:54 -08:00
event_pool.cc
…
event_pool.h
…
gpu_device.cc
Split StreamExecutor impl of pjrt_client into its own files.
2020-12-10 10:07:54 -08:00
gpu_device.h
Split StreamExecutor impl of pjrt_client into its own files.
2020-12-10 10:07:54 -08:00
gpu_multistream_test.cc
Cleanup PjRtExecutable API.
2020-12-03 15:52:14 -08:00
interpreter_device.cc
Split StreamExecutor impl of pjrt_client into its own files.
2020-12-10 10:07:54 -08:00
interpreter_device.h
Split StreamExecutor impl of pjrt_client into its own files.
2020-12-10 10:07:54 -08: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
…
pjrt_client.h
Mark Execute methods non-const allow more flexibility, e.g., lazily load programs on first execution.
2020-12-16 12:13:08 -08:00
pjrt_stream_executor_client.cc
BufferFromHostBuffer in CopyToDevice in implementation should take ownership of the intermediate literal.
2021-01-05 13:17:48 -08:00
pjrt_stream_executor_client.h
Mark Execute methods non-const allow more flexibility, e.g., lazily load programs on first execution.
2020-12-16 12:13:08 -08:00
semaphore_test.cc
…
semaphore.cc
…
semaphore.h
…
tpu_client.cc
[PJRT] Remove unneeded special case in TPU client.
2021-01-12 15:09:08 -08:00
tpu_client.h
Split StreamExecutor impl of pjrt_client into its own files.
2020-12-10 10:07:54 -08:00
tracked_device_buffer_test.cc
Remove platform field from shaped buffer.
2020-10-30 14:26:55 -07:00
tracked_device_buffer.cc
Misc fixes to PJRT impl.
2020-12-15 16:18:24 -08:00
tracked_device_buffer.h
Remove platform field from shaped buffer.
2020-10-30 14:26:55 -07:00
utils.cc
Extract helpers for PjRtClient::Compile.
2020-12-14 17:38:41 -08:00
utils.h
Extract helpers for PjRtClient::Compile.
2020-12-14 17:38:41 -08:00
worker_thread.cc
…
worker_thread.h
…