..
|
kernels
|
ops
|
tests
|
BUILD
|
build_xla_ops_pass.cc
|
build_xla_ops_pass.h
|
build_xla_ops_pass_test.cc
|
clone_constants_for_better_clustering.cc
|
clone_constants_for_better_clustering.h
|
clone_constants_for_better_clustering_test.cc
|
cluster_scoping_pass.cc
|
cluster_scoping_pass.h
|
cluster_scoping_pass_test.cc
|
compilability_check_util.cc
|
compilability_check_util.h
|
compilability_check_util_test.cc
|
compilation_passes_test_main.cc
|
deadness_analysis.cc
|
deadness_analysis.h
|
deadness_analysis_internal.h
|
deadness_analysis_test.cc
|
defs.cc
|
defs.h
|
device_compilation_cache.h
|
device_compilation_cache_test.cc
|
device_compilation_cluster_signature.cc
|
device_compilation_cluster_signature.h
|
device_compilation_cluster_signature_test.cc
|
device_compilation_profiler.cc
|
device_compilation_profiler.h
|
device_compilation_profiler_test.cc
|
device_compiler.h
|
device_compiler_client.cc
|
device_compiler_client.h
|
device_compiler_client_test.cc
|
device_compiler_disable_test.cc
|
device_compiler_test.cc
|
device_context_test.cc
|
device_executable_persistor.h
|
device_executable_persistor_test.cc
|
device_util.cc
|
device_util.h
|
device_util_test.cc
|
encapsulate_subgraphs_pass.cc
|
encapsulate_subgraphs_pass.h
|
encapsulate_subgraphs_pass_test.cc
|
encapsulate_util.cc
|
encapsulate_util.h
|
encapsulate_util_test.cc
|
encapsulate_xla_computations_pass.cc
|
encapsulate_xla_computations_pass.h
|
encapsulate_xla_computations_pass_test.cc
|
extract_outside_compilation_pass.cc
|
extract_outside_compilation_pass.h
|
extract_outside_compilation_pass_test.cc
|
flags.cc
|
flags.h
|
force_xla_constants_on_host_pass.cc
|
force_xla_constants_on_host_pass.h
|
force_xla_constants_on_host_pass_test.cc
|
get_compiler_ir.cc
|
get_compiler_ir.h
|
increase_dynamism_for_auto_jit_pass.cc
|
increase_dynamism_for_auto_jit_pass.h
|
increase_dynamism_for_auto_jit_pass_test.cc
|
jit_compilation_pass_registration.cc
|
mark_for_compilation_pass.cc
|
mark_for_compilation_pass.h
|
mark_for_compilation_pass_test.cc
|
mark_for_compilation_pass_test_helper.cc
|
mark_for_compilation_pass_test_helper.h
|
node_matchers.cc
|
node_matchers.h
|
node_matchers_test.cc
|
partially_decluster_pass.cc
|
partially_decluster_pass.h
|
partially_decluster_pass_test.cc
|
pjrt_base_device.cc
|
pjrt_base_device.h
|
pjrt_device_compiler_client.cc
|
pjrt_device_compiler_client.h
|
pjrt_device_context.cc
|
pjrt_device_context.h
|
pjrt_tensor_buffer.h
|
pjrt_tensor_buffer_util.cc
|
pjrt_tensor_buffer_util.h
|
rearrange_function_argument_pass_test.cc
|
report_clustering_info_pass.cc
|
report_clustering_info_pass.h
|
resource_operation_safety_analysis.cc
|
resource_operation_safety_analysis.h
|
resource_operation_safety_analysis_test.cc
|
shape_inference.cc
|
shape_inference.h
|
shape_inference_helpers.cc
|
shape_inference_helpers.h
|
shape_inference_test.cc
|
test_util.cc
|
test_util.h
|
tf_graph_to_hlo_compiler.cc
|
tf_graph_to_hlo_compiler.h
|
tf_to_hlo_compiler.h
|
variable_info.cc
|
variable_info.h
|
variable_info_util.cc
|
variable_info_util.h
|
xla_activity.proto
|
xla_activity_listener.cc
|
xla_activity_listener.h
|
xla_activity_listener_test.cc
|
xla_activity_logging_listener.cc
|
xla_cluster_util.cc
|
xla_cluster_util.h
|
xla_cluster_util_test.cc
|
xla_compilation_cache.proto
|
xla_compile_on_demand_op.cc
|
xla_compile_on_demand_op.h
|
xla_compile_util.cc
|
xla_compile_util.h
|
xla_compile_util_test.cc
|
xla_compiler_options_util.cc
|
xla_compiler_options_util.h
|
xla_compiler_options_util_test.cc
|
xla_cpu_device.cc
|
xla_device.cc
|
xla_device.h
|
xla_device_compiler_client.cc
|
xla_device_compiler_client.h
|
xla_device_context.cc
|
xla_device_context.h
|
xla_device_ops.cc
|
xla_device_ops.h
|
xla_gpu_device.cc
|
xla_host_recv_device_context.cc
|
xla_host_recv_device_context.h
|
xla_host_send_device_context.cc
|
xla_host_send_device_context.h
|
xla_host_send_recv_device_context_test.cc
|
xla_kernel_creator.cc
|
xla_kernel_creator.h
|
xla_kernel_creator_test.cc
|
xla_launch_util.cc
|
xla_launch_util.h
|
xla_launch_util_gpu_test.cc
|
xla_launch_util_test.cc
|
xla_ops_on_regular_devices.cc
|
xla_platform_info.cc
|
xla_platform_info.h
|
xla_platform_info_test.cc
|
xla_tensor.cc
|
xla_tensor.h
|
xla_tpu_device.cc
|
xla_tpu_device.h
|