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
365 lmhlo::ReduceWindowOp reduce_window_op, ArrayRef<Value> /*args*/, in matchAndRewrite()384 lmhlo::ReduceWindowOp reduce_window_op, in CreateParallelLoopsToTraverseOutputAndWindow()416 lmhlo::ReduceWindowOp reduce_window_op, scf::ParallelOp output_loop, in CreateReduceOpInNestedParallelLoops()
2472 auto reduce_window_op = rewriter.create<ReduceWindowOp>( in matchAndRewrite() local