Home
last modified time | relevance | path

Searched refs:reduce_window_op (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_legalize_to_parallel_loops.cc365 lmhlo::ReduceWindowOp reduce_window_op, ArrayRef<Value> /*args*/, in matchAndRewrite() argument
369 CreateParallelLoopsToTraverseOutputAndWindow(reduce_window_op, in matchAndRewrite()
373 reduce_window_op, output_loop, window_loop, &rewriter); in matchAndRewrite()
375 ConvertToReductionOperator(reduce_window_op.getLoc(), reduce_op, in matchAndRewrite()
376 &reduce_window_op.body().front(), &rewriter); in matchAndRewrite()
377 rewriter.replaceOp(reduce_window_op, llvm::None); in matchAndRewrite()
384 lmhlo::ReduceWindowOp reduce_window_op, in CreateParallelLoopsToTraverseOutputAndWindow() argument
386 auto loc = reduce_window_op.getLoc(); in CreateParallelLoopsToTraverseOutputAndWindow()
388 rewriter->create<LoadOp>(loc, reduce_window_op.init_value()); in CreateParallelLoopsToTraverseOutputAndWindow()
394 Value output = reduce_window_op.out(); in CreateParallelLoopsToTraverseOutputAndWindow()
[all …]
Dsink_constants_to_control_flow.cc53 } else if (auto reduce_window_op = llvm::dyn_cast<ReduceWindowOp>(op)) { in runOnFunction() local
54 SinkToRegion(&reduce_window_op.body()); in runOnFunction()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
DBUILD81 "reduce_window_op.cc",
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc2472 auto reduce_window_op = rewriter.create<ReduceWindowOp>( in matchAndRewrite() local
2481 BuildReduceBody<AddOp>(sum_element_type, &reduce_window_op.body(), in matchAndRewrite()
2483 Value result = reduce_window_op.getResult(); in matchAndRewrite()