Searched defs:mesh_or_status (Results 1 – 6 of 6) sorted by relevance
101 auto mesh_or_status = ExtractDeviceMeshFromOp(operand_cluster); in ExtractMeshFromOpOutput() local304 auto mesh_or_status = ExtractDeviceMeshFromOp(consumer_cluster); in InferMeshFromConsumers() local522 auto mesh_or_status = ExtractDeviceMeshFromOp(cluster); in PropagateDefaultMeshToUnAssignedClusters() local
81 auto mesh_or_status = Mesh::FromString(string(call.config())); in IdentifyTPUFunctions() local
51 auto mesh_or_status = ExtractDeviceMeshFromOp(cluster); in ExtractMeshFromCluster() local
57 auto mesh_or_status = ExtractDeviceMeshFromOp(cluster); in ExtractMeshFromCluster() local
65 auto mesh_or_status = ExtractDeviceMeshFromOp(cluster); in ExtractMeshFromCluster() local
158 auto mesh_or_status = Mesh::FromString(call_config.str()); in runOnOperation() local