Home
last modified time | relevance | path

Searched defs:mesh_dims (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Dtensor_layout.cc229 const std::string& name, const std::vector<MeshDimension>& mesh_dims) { in GetAbstractMesh()
252 const std::vector<MeshDimension>& mesh_dims, in GetMesh()
516 const std::string& name, const std::vector<MeshDimension>& mesh_dims, in GenerateMeshDevicesForTests()
569 std::vector<MeshDimension> mesh_dims; in FromString() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dreduce_spmd_expander.cc84 absl::flat_hash_set<std::string> mesh_dims; in ReducedMeshDimensions() local