Home
last modified time | relevance | path

Searched +path:tensorflow +path:dtensor +path:mlir (Results 1 – 25 of 164) sorted by relevance

1234567

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A DBUILD
H A Ddtensor_mlir_passes.cc
H A Dspmd_expander.cc
H A Dset_default_sharding.cc
H A Dvalue_utils.cc
H A Ddtensor_allreduce_scatter_optimization.cc
H A Ddce.cc
H A Ddesignate_resource_handle_mesh.cc
H A Ddtensor_send_recv.cc
H A Dhandle_sparsetensors.cc
H A DPasses.td
H A Dcollectives.cc
H A Ddtensor_mixed_precision_reduce.cc
H A Drestore_shape_inference.cc
H A Dmove_compilation_to_host.cc
H A Dsparse_expander_common.h
H A Dspmd_expanders.cc
H A Dvalue_utils.h
H A Dlayout_parsing.h
H A Ddevice_mesh_cluster_coarsening.cc
H A Dcollectives.h
H A Dcluster_function_conversion.cc
H A Ddtensor_mlir_passes_classes.h
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/dtensor_dialect/
H A DBUILD
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/utils/
H A DBUILD

1234567