Searched defs:reduce_window_op (Results 1 – 3 of 3) sorted by relevance
53 } else if (auto reduce_window_op = llvm::dyn_cast<ReduceWindowOp>(op)) { in runOnFunction() local
368 lmhlo::ReduceWindowOp reduce_window_op, ArrayRef<Value> /*args*/, in matchAndRewrite()390 lmhlo::ReduceWindowOp reduce_window_op, in CreateParallelLoopsToTraverseOutputAndWindow()423 lmhlo::ReduceWindowOp reduce_window_op, scf::ParallelOp output_loop, in CreateReduceOpInNestedParallelLoops()
3086 auto reduce_window_op = rewriter.create<ReduceWindowOp>( in matchAndRewrite() local