Google Git
Sign in
chromium / external / github.com / tensorflow / tensorflow / 034303c94a103a9290f1fe79604b6694192816e2 / . / tensorflow / compiler / xla / python
tree: 3b8101801792aea5397a73d402f28d0414d09982 [path history] [tgz]
  1. ifrt/
  2. pjrt_ifrt/
  3. profiler/
  4. tpu_driver/
  5. xla_extension/
  6. __init__.py
  7. BUILD
  8. callback.cc
  9. callback.h
  10. custom_call_for_test.pyx
  11. custom_call_sharding.cc
  12. custom_call_sharding.h
  13. dlpack.cc
  14. dlpack.h
  15. exceptions.h
  16. jax_jit.cc
  17. jax_jit.h
  18. mlir.cc
  19. mlir.h
  20. ops.cc
  21. ops.h
  22. outfeed_receiver.cc
  23. outfeed_receiver.h
  24. outfeed_receiver_py.cc
  25. outfeed_receiver_py.h
  26. outfeed_receiver_test.cc
  27. pjit.cc
  28. pjit.h
  29. pmap_lib.cc
  30. pmap_lib.h
  31. pprof_profile_builder.cc
  32. pprof_profile_builder.h
  33. profiler.cc
  34. profiler.h
  35. py_array.cc
  36. py_array.h
  37. py_buffer.cc
  38. py_buffer.h
  39. py_client.cc
  40. py_client.h
  41. py_client_gpu.cc
  42. py_client_gpu.h
  43. py_executable.cc
  44. py_executable.h
  45. py_values.cc
  46. py_values.h
  47. python_ref_manager.cc
  48. python_ref_manager.h
  49. python_utils.h
  50. pytree.cc
  51. pytree.h
  52. sharded_device_array.cc
  53. sharded_device_array.h
  54. sharding.cc
  55. sharding.h
  56. status_casters.h
  57. traceback.cc
  58. traceback.h
  59. transfer_guard_lib.cc
  60. transfer_guard_lib.h
  61. types.cc
  62. types.h
  63. util.cc
  64. util.h
  65. weakref_lru_cache.cc
  66. weakref_lru_cache.h
  67. weakref_lru_cache_test.py
  68. xla.cc
  69. xla_client.py
  70. xla_client.pyi
  71. xla_client_backend_independent_test.py
  72. xla_client_test.py
  73. xla_compiler.cc
  74. xla_compiler.h