Searched refs:branch_funcs (Results 1 – 3 of 3) sorted by relevance
368 SmallVector<GraphFuncOp> branch_funcs; in matchAndRewrite() local370 branch_funcs.reserve(op.branches().size()); in matchAndRewrite()374 branch_funcs.push_back(branch_func); in matchAndRewrite()379 for (GraphFuncOp func : branch_funcs) { in matchAndRewrite()399 SmallVector<BlockAndValueMapping> bvs(branch_funcs.size(), {}); in matchAndRewrite()402 for (auto it : llvm::zip(branch_funcs, bvs)) { in matchAndRewrite()411 for (auto it : llvm::zip(branch_funcs, region_op.branches(), bvs)) { in matchAndRewrite()
250 llvm::SmallVector<func::FuncOp, 4> branch_funcs; in CollectOpSideEffects() local251 case_op.get_branch_functions(branch_funcs); in CollectOpSideEffects()252 for (auto branch_func : branch_funcs) { in CollectOpSideEffects()
558 llvm::SmallVector<mlir::func::FuncOp> branch_funcs( in ExpandRestoreV2OpHelper() local631 branch_funcs[local_device_idx] = device_restore_fn; in ExpandRestoreV2OpHelper()636 for (auto& func : branch_funcs) in ExpandRestoreV2OpHelper()