backward compat: Prevent changes to attr's default value only for last historical version.

PiperOrigin-RevId: 181646665

1 files changed, 9 insertions, 1 deletions

diff --git a/tensorflow/core/framework/op_compatibility_test.cc b/tensorflow/core/framework/op_compatibility_test.cc
index 4f4813d9fa..b57bdcb841 100644
--- a/tensorflow/core/framework/op_compatibility_test.cc
+++ b/tensorflow/core/framework/op_compatibility_test.cc
@@ -173,7 +173,15 @@ class OpCompatibilityTest : public OpsTestBase {
     // Validate that the NodeDef is valid.
     TF_ASSERT_OK(ValidateNodeDef(*node_def(), *new_op_def));
 
-    ExpectIncompatible(old_op_def, *new_op_def, compatibility_error);
+    Status status = OpDefAttrDefaultsUnchanged(old_op_def, *new_op_def);
+    if (status.ok()) {
+      ADD_FAILURE() << SummarizeOpDef(old_op_def) << " vs. "
+                    << SummarizeOpDef(*new_op_def);
+    } else {
+      EXPECT_TRUE(
+          StringPiece(status.error_message()).contains(compatibility_error))
+          << status << " does not contain " << compatibility_error;
+    }
   }
 };