Add regression test demonstrating a codegen error.
FoldInitTensorWithTensorReshapeOp rewrite pattern runs into a segfault.
PiperOrigin-RevId: 420974329
Change-Id: I51adbbd5329bb00b5162c7cb998cffeeeae7a793
diff --git a/tensorflow/compiler/mlir/tfrt/python_tests/regression_tests/broadcast_of_squeeze.disabled.mlir b/tensorflow/compiler/mlir/tfrt/python_tests/regression_tests/broadcast_of_squeeze.disabled.mlir
new file mode 100644
index 0000000..a358ad2
--- /dev/null
+++ b/tensorflow/compiler/mlir/tfrt/python_tests/regression_tests/broadcast_of_squeeze.disabled.mlir
@@ -0,0 +1,6 @@
+builtin.func @test(%V__0 : tensor<?x1x?xi64> { python_test_attrs.static_type = tensor<1x1x1xi64> }, %V__1 : tensor<?x?x1xi1> { python_test_attrs.static_type = tensor<1x60x1xi1> }) -> tensor<?x?x?xi64> {
+ %0 = "tf.Squeeze"(%V__0) { squeeze_dims = [ 0 : i64, 2 : i64 ] } : (tensor<?x1x?xi64>) -> tensor<1xi64>
+ %1 = "tf.Shape"(%V__1) : (tensor<?x?x1xi1>) -> tensor<3xi32>
+ %2 = "tf.BroadcastTo"(%0, %1) : (tensor<1xi64>, tensor<3xi32>) -> tensor<?x?x?xi64>
+ return %2 : tensor<?x?x?xi64>
+}
