Home
last modified time | relevance | path

Searched defs:mesh_or (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_sharded_prefix_kernel.cc60 const auto& mesh_or = Mesh::FromString(mesh_str); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/dtensor_dialect/ir/
H A Dops.cc72 MeshOr mesh_or = Mesh::FromString(mesh_str); in ParseMeshAttr() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dmove_compilation_to_host.cc57 auto mesh_or = ExtractDeviceMeshFromOp(dtensor_function); in IdentifyAndValidateMeshComputations() local