Searched refs:function_mesh (Results 1 – 4 of 4) sorted by relevance
245 absl::optional<Mesh> function_mesh; in ExtractMeshFromFuctionOutput() local252 function_mesh.emplace(layout->mesh()); in ExtractMeshFromFuctionOutput()253 return function_mesh; in ExtractMeshFromFuctionOutput()261 function_mesh.emplace(std::move(mesh)); in ExtractMeshFromFuctionOutput()263 return function_mesh; in ExtractMeshFromFuctionOutput()591 absl::optional<mlir::StringAttr> function_mesh; in PropagateMeshFromInputs() local593 builder, &function_mesh))) in PropagateMeshFromInputs()596 if (function_mesh && !cluster_mesh) { in PropagateMeshFromInputs()598 cluster->setAttr(kMeshAttr, function_mesh.value()); in PropagateMeshFromInputs()638 absl::optional<mlir::StringAttr> function_mesh; in PropagateMeshFromConsumers() local[all …]
1447 function.function_mesh.GlobalFingerprint()); in LowerToSPMDFunction()1460 const std::string mesh_str = function_ptr->function_mesh.ToString(); in ExecuteFunctionAndWait()1531 StatusOr<Mesh> maybe_converted_mesh = function.function_mesh; in ExecuteRegularOperation()1532 if (function.function_mesh.is_epu_mesh()) { in ExecuteRegularOperation()1533 maybe_converted_mesh = function.function_mesh.ToDeviceType("CPU"); in ExecuteRegularOperation()1556 if (function.function_mesh.is_epu_mesh()) { in ExecuteRegularOperation()1643 const Mesh& mesh = function.function_mesh; in ExecuteRegularOperation()1705 const Mesh& mesh = function.function_mesh; in ExecuteRegularOperation()
71 Mesh function_mesh; member
860 function.function_mesh = std::move(mesh); in IdentifyAllFunctionsToExecute()