| .. |
|
tpu_driver
|
Add check for events_ not containing the event we are waiting on because it has already completed
|
2020-10-14 15:39:09 -07:00 |
|
__init__.py
|
|
|
|
bfloat16_test.py
|
[XLA:Python] Fix bug where bfloat16 dtype extension's hash value was not initialized correctly.
|
2020-10-20 10:49:10 -07:00 |
|
bfloat16.cc
|
[XLA:Python] Fix bug where bfloat16 dtype extension's hash value was not initialized correctly.
|
2020-10-20 10:49:10 -07:00 |
|
bfloat16.h
|
systemlibs: unbundle pybind11
|
2020-04-08 03:38:38 +08:00 |
|
BUILD
|
[XLA:Python] Add support for DLPack buffer sharing without taking ownership.
|
2020-10-20 06:28:31 -07:00 |
|
custom_call_for_test.pyx
|
[XLA:Python] Add NextAfter to Python client.
|
2019-10-21 11:20:02 -07:00 |
|
dlpack.cc
|
[XLA:Python] Add support for DLPack buffer sharing without taking ownership.
|
2020-10-20 06:28:31 -07:00 |
|
dlpack.h
|
[XLA:Python] Add support for DLPack buffer sharing without taking ownership.
|
2020-10-20 06:28:31 -07:00 |
|
jax_jit.cc
|
Refactor PJRT.
|
2020-10-20 11:57:11 -07:00 |
|
jax_jit.h
|
Add a C++ jitlib library targeting Jax.
|
2020-08-14 12:12:51 -07:00 |
|
ops.cc
|
Add XLA registration for Zeta/Zetac
|
2020-09-18 13:04:32 -07:00 |
|
ops.h
|
[XLA:Python] Split bindings for XLA ops into a separate file. No functional changes.
|
2020-05-26 18:38:07 -07:00 |
|
outfeed_receiver_py.cc
|
[PJRT] Rename Device to PjRtDevice for consistency with the other PjRt classes.
|
2020-08-19 11:58:30 -07:00 |
|
outfeed_receiver_py.h
|
An implementation of a multithreaded runtime for receiving outfeed data
|
2020-05-27 05:52:21 -07:00 |
|
outfeed_receiver_test.cc
|
Refactor PJRT.
|
2020-10-20 11:57:11 -07:00 |
|
outfeed_receiver.cc
|
Refactor PJRT.
|
2020-10-20 11:57:11 -07:00 |
|
outfeed_receiver.h
|
[PJRT] Rename Device to PjRtDevice for consistency with the other PjRt classes.
|
2020-08-19 11:58:30 -07:00 |
|
py_buffer.cc
|
[PJRT] Rename Device to PjRtDevice for consistency with the other PjRt classes.
|
2020-08-19 11:58:30 -07:00 |
|
py_buffer.h
|
[PJRT] Rename Device to PjRtDevice for consistency with the other PjRt classes.
|
2020-08-19 11:58:30 -07:00 |
|
py_client.cc
|
Refactor PJRT.
|
2020-10-20 11:57:11 -07:00 |
|
py_client.h
|
Refactor PJRT.
|
2020-10-20 11:57:11 -07:00 |
|
py_executable.cc
|
[PJRT] Rename Device to PjRtDevice for consistency with the other PjRt classes.
|
2020-08-19 11:58:30 -07:00 |
|
py_executable.h
|
[PJRT] Rename Device to PjRtDevice for consistency with the other PjRt classes.
|
2020-08-19 11:58:30 -07:00 |
|
python_ref_manager.cc
|
[XLA:Python] Optimize traceback implementation.
|
2020-06-05 18:42:03 -07:00 |
|
python_ref_manager.h
|
[XLA:Python] Optimize traceback implementation.
|
2020-06-05 18:42:03 -07:00 |
|
pytree.cc
|
Fix some memory issue.
|
2020-08-17 09:44:56 -07:00 |
|
pytree.h
|
Fix some memory issue.
|
2020-08-17 09:44:56 -07:00 |
|
traceback.cc
|
[XLA:Python] Implement on-device heap profiling support for JAX.
|
2020-06-12 11:18:13 -07:00 |
|
traceback.h
|
[XLA:Python] Implement on-device heap profiling support for JAX.
|
2020-06-12 11:18:13 -07:00 |
|
types.cc
|
[XLA:Python] Implement __cuda_array_interface__ on DeviceBuffer.
|
2020-01-30 17:45:31 -08:00 |
|
types.h
|
[XLA:Python] Add a PyClient wrapper object around PjRtClient.
|
2020-06-05 14:18:04 -07:00 |
|
xla_client_backend_independent_test.py
|
[XLA:Python] Validate shapes in Python bindings to avoid crashes.
|
2020-10-15 13:20:25 -07:00 |
|
xla_client_test.py
|
[XLA:Python] Add support for DLPack buffer sharing without taking ownership.
|
2020-10-20 06:28:31 -07:00 |
|
xla_client.py
|
[XLA:Python] Add TPU backend.
|
2020-10-14 16:07:01 -07:00 |
|
xla.cc
|
Refactor PJRT.
|
2020-10-20 12:08:59 -07:00 |
c98ac56bc2