Home
last modified time | relevance | path

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

123456789

/external/tensorflow/tensorflow/dtensor/mlir/
Dop_utils.cc
Ddtensor_mlir_passes.h
Ddevice_mesh_cluster_coarsening.cc
Dcollectives_common.cc
Dtpu_add_resource_device_attribute.cc
Ddesignate_resource_handle_mesh.cc
Dlayout_parsing.cc
Dundo_merge_const_across_mesh.cc
Dlower_send_recv.cc
Dgroup_assignment_test.cc
Dmerge_clusters.cc
Dsparse_expander.cc
Ddtensor_mlir_passes.cc
Dsparse_expansion.cc
Ddtensor_allreduce_sum_optimization.cc
Dsparse_expanders.cc
Dcollectives.h
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_graph_to_mlir_pass.h
Ddtensor_device_util.h
Dsave_restore_util.h
Ddtensor_sharded_prefix_kernel.cc
/external/tensorflow/tensorflow/dtensor/python/
Dlayout.py
D__init__.py
Dtpu_util.py
/external/tensorflow/tensorflow/dtensor/tests/
Dtensor_layout_test.cc

123456789