tree: 294bf41a7a813fb2bab96696f6c0057673a56c59 [path history] [tgz]
  1. xla/
  2. BUILD
  3. compiled_subgraph.h
  4. cross_replica_ops.cc
  5. host_compute_ops.cc
  6. image_resize_ops.cc
  7. infeed_ops.cc
  8. infeed_ops.h
  9. outfeed_ops.cc
  10. outfeed_ops.h
  11. replication_ops.cc
  12. sharding_util_ops.cc
  13. sharding_util_ops_test.cc
  14. topk_ops.cc
  15. tpu_compilation_cache.proto
  16. tpu_compilation_cache_common.proto
  17. tpu_compilation_cache_entry.h
  18. tpu_compilation_cache_entry_unloader.h
  19. tpu_compilation_cache_external.cc
  20. tpu_compilation_cache_external.h
  21. tpu_compilation_cache_factory.cc
  22. tpu_compilation_cache_factory.h
  23. tpu_compilation_cache_grpc.cc
  24. tpu_compilation_cache_grpc.h
  25. tpu_compilation_cache_interface.cc
  26. tpu_compilation_cache_interface.h
  27. tpu_compilation_cache_key.h
  28. tpu_compilation_cache_local_lookup.cc
  29. tpu_compilation_cache_local_lookup.h
  30. tpu_compilation_cache_lookup.h
  31. tpu_compilation_cache_rpc_lookup.cc
  32. tpu_compilation_cache_rpc_lookup.h
  33. tpu_compilation_cache_rpc_support.cc
  34. tpu_compilation_cache_rpc_support.h
  35. tpu_compilation_cache_service.cc
  36. tpu_compilation_cache_service.h
  37. tpu_compilation_metrics.cc
  38. tpu_compilation_metrics.h
  39. tpu_compile.proto
  40. tpu_compile_op.cc
  41. tpu_compile_op.h
  42. tpu_compile_op_common.cc
  43. tpu_compile_op_common.h
  44. tpu_compile_op_impl.cc
  45. tpu_compile_op_impl.h
  46. tpu_compile_op_options.cc
  47. tpu_compile_op_options.h
  48. tpu_compile_op_support.cc
  49. tpu_compile_op_support.h
  50. tpu_configuration_ops.cc
  51. tpu_configuration_ops.h
  52. tpu_executable_info.proto
  53. tpu_execute_op.cc
  54. tpu_execute_op.h
  55. tpu_fingerprint_lookup.cc
  56. tpu_fingerprint_lookup.h
  57. tpu_functional_ops.cc
  58. tpu_functional_ops.h
  59. tpu_handle_to_key_op.cc
  60. tpu_mesh_state_interface.h
  61. tpu_op_consts.cc
  62. tpu_op_consts.h
  63. tpu_op_util.cc
  64. tpu_op_util.h
  65. tpu_ordinal_selector.h
  66. tpu_ordinal_selector_interface.h
  67. tpu_ordinal_selector_op.cc
  68. tpu_pod_state.cc
  69. tpu_pod_state.h
  70. tpu_program_group.cc
  71. tpu_program_group.h
  72. tpu_program_group_interface.h
  73. tpu_reshard_variables_op.cc
  74. tpu_reshard_variables_op.h
  75. tpu_reshard_variables_op_util.cc
  76. tpu_reshard_variables_op_util.h
  77. tpu_util.cc
  78. tpu_util.h
  79. trace_util.h
  80. transfer_ops.cc
  81. transfer_ops.h