Google Git
Sign in
chromium/external/github.com/tensorflow/tensorflow/refs/heads/master/./third_party/xla/xla/pjrt
tree: f3c9f132a604ace71a74a793626cceb7250274ed
  1. c/
  2. c_api_client/
  3. cpu/
  4. distributed/
  5. dump/
  6. extensions/
  7. gpu/
  8. interpreter/
  9. plugin/
  10. profiling/
  11. proto/
  12. abstract_tracked_device_buffer.cc
  13. abstract_tracked_device_buffer.h
  14. async_work_runner.h
  15. buffer_sequencing_event.cc
  16. buffer_sequencing_event.h
  17. BUILD
  18. common_pjrt_client.cc
  19. common_pjrt_client.h
  20. compiled_memory_stats.cc
  21. compiled_memory_stats.h
  22. device_event.h
  23. errors.cc
  24. errors.h
  25. errors_test.cc
  26. event_pool.cc
  27. event_pool.h
  28. exceptions.h
  29. host_callback.cc
  30. host_callback.h
  31. host_callback_test.cc
  32. host_memory_allocator.cc
  33. host_memory_allocator.h
  34. host_memory_spaces.cc
  35. host_memory_spaces.h
  36. host_to_device_transfer_manager.cc
  37. host_to_device_transfer_manager.h
  38. layout_mode.cc
  39. layout_mode.h
  40. local_device_state.cc
  41. local_device_state.h
  42. lru_cache.h
  43. lru_cache_test.cc
  44. maybe_owning_mlir_module.h
  45. metrics.cc
  46. metrics.h
  47. mlir_to_hlo.cc
  48. mlir_to_hlo.h
  49. mlir_to_hlo_test.cc
  50. never_run_on_fiber.h
  51. partial_program_utils.cc
  52. partial_program_utils.h
  53. pjrt_abi_version.h
  54. pjrt_api.cc
  55. pjrt_api.h
  56. pjrt_api_test.cc
  57. pjrt_c_api_client.h
  58. pjrt_client.cc
  59. pjrt_client.h
  60. pjrt_client_test.cc
  61. pjrt_client_test.h
  62. pjrt_client_utils.cc
  63. pjrt_client_utils.h
  64. pjrt_common.cc
  65. pjrt_common.h
  66. pjrt_common_test.cc
  67. pjrt_compiler.cc
  68. pjrt_compiler.h
  69. pjrt_compiler_test.cc
  70. pjrt_device_description.h
  71. pjrt_device_dimensions.cc
  72. pjrt_device_dimensions.h
  73. pjrt_device_dimensions_test.cc
  74. pjrt_executable.cc
  75. pjrt_executable.h
  76. pjrt_executable_test.cc
  77. pjrt_future.h
  78. pjrt_layout.h
  79. pjrt_phase_compile_extension_test.cc
  80. pjrt_phase_compile_sample_plugin.cc
  81. pjrt_phase_compile_sample_plugin.h
  82. pjrt_phase_compiler_test.cc
  83. pjrt_stream_executor_client.cc
  84. pjrt_stream_executor_client.h
  85. pjrt_stream_executor_client_test.cc
  86. pjrt_stream_executor_device_description.h
  87. raw_buffer.cc
  88. raw_buffer.h
  89. raw_buffer_test.cc
  90. scoped_async_tracking_event.h
  91. se_raw_buffer.cc
  92. se_raw_buffer.h
  93. semaphore.cc
  94. semaphore.h
  95. semaphore_test.cc
  96. status_casters.h
  97. stream_executor_executable.cc
  98. stream_executor_executable.h
  99. stream_executor_executable.proto
  100. stream_executor_pjrt_abi_version.cc
  101. stream_executor_pjrt_abi_version.h
  102. stream_executor_pjrt_abi_version_test.cc
  103. stream_executor_platform_id_mapping.cc
  104. stream_executor_platform_id_mapping.h
  105. stream_executor_platform_id_mapping_test.cc
  106. string_utils.cc
  107. string_utils.h
  108. tf_pjrt_client.cc
  109. tf_pjrt_client.h
  110. tf_pjrt_client_test.cc
  111. tfrt_cpu_pjrt_client.h
  112. thread_pool_async_work_runner.cc
  113. thread_pool_async_work_runner.h
  114. tpu_constants.h
  115. tracked_device_buffer.cc
  116. tracked_device_buffer.h
  117. tracked_device_buffer_test.cc
  118. transpose.cc
  119. transpose.h
  120. transpose_kernels.h
  121. transpose_test.cc
  122. triton.h
  123. triton_cuda.cc
  124. triton_rocm.cc
  125. triton_stub.cc
  126. utils.cc
  127. utils.h
  128. worker_thread.cc
  129. worker_thread.h