Google Git
Sign in
chromium / external / github.com / tensorflow / tensorflow / 890252874c02e778d21bdb5779ed19e27dd2d365 / . / tensorflow / compiler / xla / python
tree: da8f55279e9667e66902bd30c17b0dec3ced31cb [path history] [tgz]
  1. tpu_driver/
  2. xla_extension/
  3. __init__.py
  4. bfloat16.cc
  5. bfloat16.h
  6. BUILD
  7. callback.cc
  8. callback.h
  9. custom_call_for_test.pyx
  10. custom_call_sharding.cc
  11. custom_call_sharding.h
  12. dlpack.cc
  13. dlpack.h
  14. exceptions.h
  15. float8_e4m3b11.cc
  16. float8_e4m3b11.h
  17. jax_jit.cc
  18. jax_jit.h
  19. mlir.cc
  20. mlir.h
  21. numpy.cc
  22. numpy.h
  23. ops.cc
  24. ops.h
  25. outfeed_receiver.cc
  26. outfeed_receiver.h
  27. outfeed_receiver_py.cc
  28. outfeed_receiver_py.h
  29. outfeed_receiver_test.cc
  30. pjit.cc
  31. pjit.h
  32. pmap_lib.cc
  33. pmap_lib.h
  34. pprof_profile_builder.cc
  35. pprof_profile_builder.h
  36. profiler.cc
  37. profiler.h
  38. py_array.cc
  39. py_array.h
  40. py_buffer.cc
  41. py_buffer.h
  42. py_client.cc
  43. py_client.h
  44. py_client_gpu.cc
  45. py_client_gpu.h
  46. py_executable.cc
  47. py_executable.h
  48. py_values.cc
  49. py_values.h
  50. python_ref_manager.cc
  51. python_ref_manager.h
  52. python_utils.h
  53. pytree.cc
  54. pytree.h
  55. sharded_device_array.cc
  56. sharded_device_array.h
  57. sharding.cc
  58. sharding.h
  59. status_casters.h
  60. traceback.cc
  61. traceback.h
  62. transfer_guard_lib.cc
  63. transfer_guard_lib.h
  64. types.cc
  65. types.h
  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