Google Git
Sign in
chromium/external/github.com/tensorflow/tensorflow/refs/heads/master/./third_party/xla/xla/pjrt/gpu
tree: a503e1275ad8c19ccb6f51c3e800c73eac391854
  1. tfrt/
  2. BUILD
  3. gpu_helpers.cc
  4. gpu_helpers.h
  5. gpu_metrics.cc
  6. gpu_metrics.h
  7. nvshmem_gpu_collectives_test.cc
  8. package_groups.bzl
  9. pjrt_client_test_se_gpu.cc
  10. se_gpu_pjrt_client.cc
  11. se_gpu_pjrt_client.h
  12. se_gpu_pjrt_client_benchmark_test.cc
  13. se_gpu_pjrt_client_nvshmem_test.cc
  14. se_gpu_pjrt_client_test.cc
  15. se_gpu_pjrt_compiler.cc
  16. se_gpu_pjrt_compiler.h
  17. se_gpu_pjrt_compiler_aot_test.cc
  18. se_gpu_pjrt_compiler_cuda_registration.cc
  19. se_gpu_pjrt_compiler_rocm_registration.cc
  20. se_gpu_pjrt_compiler_test.cc
  21. se_gpu_pjrt_runtime_abi_version.cc
  22. se_gpu_pjrt_runtime_abi_version.h
  23. se_gpu_pjrt_runtime_abi_version_test.cc
  24. se_gpu_topology_description.cc
  25. se_gpu_topology_description.h
  26. se_gpu_topology_description_test.cc