Home
last modified time | relevance | path

Searched defs:mesh_or_status (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dmesh_propagation.cc101 auto mesh_or_status = ExtractDeviceMeshFromOp(operand_cluster); in ExtractMeshFromOpOutput() local
304 auto mesh_or_status = ExtractDeviceMeshFromOp(consumer_cluster); in InferMeshFromConsumers() local
522 auto mesh_or_status = ExtractDeviceMeshFromOp(cluster); in PropagateDefaultMeshToUnAssignedClusters() local
H A Dtpu_integration.cc81 auto mesh_or_status = Mesh::FromString(string(call.config())); in IdentifyTPUFunctions() local
H A Dlower_send_recv.cc51 auto mesh_or_status = ExtractDeviceMeshFromOp(cluster); in ExtractMeshFromCluster() local
H A Dhandle_cross_cluster_dependencies.cc57 auto mesh_or_status = ExtractDeviceMeshFromOp(cluster); in ExtractMeshFromCluster() local
H A Dmerge_clusters.cc65 auto mesh_or_status = ExtractDeviceMeshFromOp(cluster); in ExtractMeshFromCluster() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/utils/
H A Dupdate_tpu_metadata.cc158 auto mesh_or_status = Mesh::FromString(call_config.str()); in runOnOperation() local