Google Git
Sign in
chromium / external / github.com / tensorflow / tensorflow / f4df0f6f2c8a559db4c321c764cbfd8d567bc552 / . / tensorflow / compiler / xla / python
tree: 724fdc580dfa6724cffbf7e0751518171d3a9a47 [path history] [tgz]
  1. profiler/
  2. tpu_driver/
  3. xla_extension/
  4. __init__.py
  5. BUILD
  6. callback.cc
  7. callback.h
  8. custom_call_for_test.pyx
  9. custom_call_sharding.cc
  10. custom_call_sharding.h
  11. dlpack.cc
  12. dlpack.h
  13. exceptions.h
  14. jax_jit.cc
  15. jax_jit.h
  16. mlir.cc
  17. mlir.h
  18. numpy.cc
  19. numpy.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.h
  64. weakref_lru_cache.cc
  65. weakref_lru_cache.h
  66. xla.cc
  67. xla_client.py
  68. xla_client.pyi
  69. xla_client_backend_independent_test.py
  70. xla_client_test.py
  71. xla_compiler.cc
  72. xla_compiler.h