Extends forward-compat window (again) for AssignVariableOp runtime shape check.
PiperOrigin-RevId: 424425087
Change-Id: I8a77c3238aa2d7a3a7d99c0660e817b0fa322302
diff --git a/tensorflow/python/kernel_tests/variables/resource_variable_ops_test.py b/tensorflow/python/kernel_tests/variables/resource_variable_ops_test.py
index de073c3..d4e3e35 100644
--- a/tensorflow/python/kernel_tests/variables/resource_variable_ops_test.py
+++ b/tensorflow/python/kernel_tests/variables/resource_variable_ops_test.py
@@ -826,7 +826,7 @@
self.assertEqual(4.0, self.evaluate(v.value()))
def testAssignRuntimeShapeCheck(self):
- with forward_compat.forward_compatibility_horizon(2022, 2, 1):
+ with forward_compat.forward_compatibility_horizon(2022, 3, 2):
v = resource_variable_ops.ResourceVariable([1.0, 1.0], name="var0")
@def_function.function
diff --git a/tensorflow/python/ops/resource_variable_ops.py b/tensorflow/python/ops/resource_variable_ops.py
index 4421528..d0e3c7a 100644
--- a/tensorflow/python/ops/resource_variable_ops.py
+++ b/tensorflow/python/ops/resource_variable_ops.py
@@ -933,7 +933,7 @@
f"The variable shape {self._shape}, and the "
f"assigned value shape {value_tensor.shape} are incompatible."))
kwargs = {}
- if forward_compat.forward_compatible(2022, 1, 27):
+ if forward_compat.forward_compatible(2022, 2, 23):
# If the shape is fully defined, we do a runtime check with the shape of
# value.
validate_shape = self._shape.is_fully_defined()
