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