Add regression test that fails with vectorization turned on.
PiperOrigin-RevId: 410473208
Change-Id: Id8650c24a18dd4c415cf4180b85a51e1a35e1180
diff --git a/tensorflow/compiler/mlir/tfrt/python_tests/regression_tests/reduction_of_reduction.mlir b/tensorflow/compiler/mlir/tfrt/python_tests/regression_tests/reduction_of_reduction.mlir
new file mode 100644
index 0000000..f854262
--- /dev/null
+++ b/tensorflow/compiler/mlir/tfrt/python_tests/regression_tests/reduction_of_reduction.mlir
@@ -0,0 +1,13 @@
+builtin.func @test(%V__0: tensor<?x?x?xi64>
+ { python_test_attrs.static_type = tensor<9x32x26xi64> })
+ -> tensor<?x?xi64> {
+ %dims1 = "tf.Const"() { value = dense<[0]> : tensor<1xi32> }:
+ () -> tensor<1xi32>
+ %1 = "tf.Mean"(%V__0, %dims1) {keep_dims = false} :
+ (tensor<?x?x?xi64>, tensor<1xi32>) -> tensor<?x?xi64>
+ %dims2 = "tf.Const"() { value = dense<[1]> : tensor<1xi32> }:
+ () -> tensor<1xi32>
+ %2 = "tf.Min"(%1, %dims2) {keep_dims = true} :
+ (tensor<?x?xi64>, tensor<1xi32>) -> tensor<?x?xi64>
+ return %2 : tensor<?x?xi64>
+}
