|
..
|
|
dtensor_dialect
|
|
expansions
|
|
ir
|
|
sparse_expansions
|
|
tests
|
|
utils
|
|
BUILD
|
|
Passes.td
|
|
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_layout_to_xla_sharding_op.cc
|
|
dtensor_location.cc
|
|
dtensor_location.h
|
|
dtensor_location_test.cc
|
|
dtensor_mixed_precision_reduce.cc
|
|
dtensor_mlir_passes.cc
|
|
dtensor_mlir_passes.h
|
|
dtensor_multi_device_expansion.cc
|
|
dtensor_remove_dtensorlayout.cc
|
|
dtensor_replace_auxiliary_layout_op.cc
|
|
dtensor_replace_relayout_with_identity.cc
|
|
dtensor_send_recv.cc
|
|
dtensor_send_recv.h
|
|
dtensor_set_hlo_sharding.cc
|
|
function_renaming.cc
|
|
group_assignment.cc
|
|
group_assignment.h
|
|
group_assignment_test.cc
|
|
handle_cross_cluster_dependencies.cc
|
|
handle_sparsetensors.cc
|
|
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
|
|
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
|
|
value_utils.cc
|
|
value_utils.h
|