Searched refs:xla_variadic_reduce_v2_op (Results 1 – 2 of 2) sorted by relevance
2441 } else if (auto xla_variadic_reduce_v2_op = in PropagateShapeIntoAttachedFunctions() local2443 return propagate_shape_to(xla_variadic_reduce_v2_op.reducer()); in PropagateShapeIntoAttachedFunctions()
3657 mlir::TF::XlaVariadicReduceV2Op xla_variadic_reduce_v2_op = in matchAndRewrite() local3662 rewriter.replaceOp(op, xla_variadic_reduce_v2_op.getResults()); in matchAndRewrite()