Google Git
Sign in
chromium / external / github.com / tensorflow / tensorflow / 02818438679d4f691f54dc640a7b5cb3438edf30 / . / tensorflow / compiler / xla / python
tree: 29814a12db6744aacbfeee8c60765f8612612351 [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. numpy.cc
  21. numpy.h
  22. ops.cc
  23. ops.h
  24. outfeed_receiver.cc
  25. outfeed_receiver.h
  26. outfeed_receiver_py.cc
  27. outfeed_receiver_py.h
  28. outfeed_receiver_test.cc
  29. pjit.cc
  30. pjit.h
  31. pmap_lib.cc
  32. pmap_lib.h
  33. pprof_profile_builder.cc
  34. pprof_profile_builder.h
  35. profiler.cc
  36. profiler.h
  37. py_array.cc
  38. py_array.h
  39. py_buffer.cc
  40. py_buffer.h
  41. py_client.cc
  42. py_client.h
  43. py_client_gpu.cc
  44. py_client_gpu.h
  45. py_executable.cc
  46. py_executable.h
  47. py_values.cc
  48. py_values.h
  49. python_ref_manager.cc
  50. python_ref_manager.h
  51. python_utils.h
  52. pytree.cc
  53. pytree.h
  54. sharded_device_array.cc
  55. sharded_device_array.h
  56. sharding.cc
  57. sharding.h
  58. status_casters.h
  59. traceback.cc
  60. traceback.h
  61. transfer_guard_lib.cc
  62. transfer_guard_lib.h
  63. types.cc
  64. types.h
  65. util.cc
  66. util.h
  67. weakref_lru_cache.cc
  68. weakref_lru_cache.h
  69. xla.cc
  70. xla_client.py
  71. xla_client.pyi
  72. xla_client_backend_independent_test.py
  73. xla_client_test.py
  74. xla_compiler.cc
  75. xla_compiler.h