../
annotate_global_shape.cc
cluster_function_conversion.cc
collectives.cc
collectives.h
collectives_common.cc
collectives_common.h
constant_folding.cc
create_dtensor_mlir_passes.h
dce.cc
designate_resource_handle_mesh.cc
device_mesh_cluster_coarsening.cc
device_utils.cc
device_utils.h
dtensor_allreduce_combine_optimization.cc
dtensor_allreduce_scatter_optimization.cc
dtensor_allreduce_sum_optimization.cc
[+] dtensor_dialect/
dtensor_location.cc
dtensor_location.h
dtensor_mixed_precision_reduce.cc
dtensor_mlir_passes.cc
dtensor_mlir_passes.h
dtensor_passes.h.inc
dtensor_send_recv.cc
dtensor_send_recv.h
[+] expansions/
function_renaming.cc
group_assignment.cc
group_assignment.h
handle_cross_cluster_dependencies.cc
handle_sparsetensors.cc
[+] ir/
layout_parsing.cc
layout_parsing.h
layout_propagation_v2.cc
lower_send_recv.cc
merge_clusters.cc
mesh_propagation.cc
move_compilation_to_host.cc
op_to_device_cluster.cc
op_utils.cc
op_utils.h
propagate_default_layout.cc
propagate_device_id_to_function_args.cc
restore_shape_inference.cc
set_default_sharding.cc
shape_utils.cc
shape_utils.h
sparse_expander.cc
sparse_expander.h
sparse_expander_common.cc
sparse_expander_common.h
sparse_expanders.cc
sparse_expansion.cc
[+] sparse_expansions/
spmd_expander.cc
spmd_expander.h
spmd_expander_common.cc
spmd_expander_common.h
spmd_expanders.cc
spmd_expansion.cc
tpu_add_resource_device_attribute.cc
tpu_integration.cc
undo_merge_const_across_mesh.cc
[+] utils/
value_utils.cc
value_utils.h