Browse the source code of
tensorflow
/
tensorflow
/
tensorflow
/
dtensor
/
mlir/
online
../
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
DTensorDeviceMeshClusterCoarsening
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
tensorflow::dtensor::(anonymous namespace)::impl::DTensorAnnotateGlobalShapeBase
dtensor_send_recv.cc
dtensor_send_recv.h
[+]
expansions/
function_renaming.cc
group_assignment.cc
group_assignment.h
GroupAssignment
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
SparseExpanderBase
SparseExpanderRegistry
sparse_expander_common.cc
sparse_expander_common.h
sparse_expanders.cc
sparse_expansion.cc
[+]
sparse_expansions/
spmd_expander.cc
spmd_expander.h
SPMDExpanderBase
SPMDExpanderRegistry
spmd_expander_common.cc
spmd_expander_common.h
TopologicalIterator
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
Generated on
2022-Sep-26
Powered by
Code Browser
2.1
Generator usage only permitted with license