Searched refs:window_loop (Results 1 – 1 of 1) sorted by relevance
367 scf::ParallelOp output_loop, window_loop; in matchAndRewrite() local368 std::tie(output_loop, window_loop) = in matchAndRewrite()373 reduce_window_op, output_loop, window_loop, &rewriter); in matchAndRewrite()406 auto window_loop = rewriter->create<scf::ParallelOp>( in CreateParallelLoopsToTraverseOutputAndWindow() local409 Value reduction_result = *window_loop.getResults().begin(); in CreateParallelLoopsToTraverseOutputAndWindow()412 return std::make_pair(output_loop, window_loop); in CreateParallelLoopsToTraverseOutputAndWindow()417 scf::ParallelOp window_loop, ConversionPatternRewriter* rewriter) const { in CreateReduceOpInNestedParallelLoops() argument418 rewriter->setInsertionPointToStart(window_loop.getBody()); in CreateReduceOpInNestedParallelLoops()434 window_loop.getInductionVars(), rewriter); in CreateReduceOpInNestedParallelLoops()448 else_builder.create<scf::YieldOp>(loc, *window_loop.initVals().begin()); in CreateReduceOpInNestedParallelLoops()