Searched refs:BuildReduceBody (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 572 static void BuildReduceBody(Type element_type, Region *body, in BuildReduceBody() function
1415 BuildReduceBody<AddOp>(input_type.getElementType(), &reduce.body(), in matchAndRewrite()
2241 BuildReduceBody<AddOp>(element_type, &divisor.body(), &rewriter); in AvgPoolDivideByCount()
2295 BuildReduceBody<AddOp>(sum_element_type, &reduce.body(), &rewriter); in matchAndRewrite()
2481 BuildReduceBody<AddOp>(sum_element_type, &reduce_window_op.body(), in matchAndRewrite()
2537 BuildReduceBody<MaxOp>(element_type, &reduce.body(), &rewriter); in matchAndRewrite()
3668 BuildReduceBody<ReductionOp>(reduce_element_type, &reduction.body(), in matchAndRewrite()
4068 BuildReduceBody<AddOp>(element_type, &result.scatter(), &rewriter); in matchAndRewrite()
4915 BuildReduceBody<ReductionOp>(data_type.getElementType(), in matchAndRewrite()
5278 BuildReduceBody<AddOp>(element_type, &all_reduce.computation(), in matchAndRewrite()
[all …]
|