experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/compiler/xla/python
History
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
..
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
…
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