tf.assert_greater_equal is now tf.compat.v1.assert_greater_equal
PiperOrigin-RevId: 277316693
Change-Id: Ib203238734aa34261ad25093ca35f0b3deea0ed7
diff --git a/tensorflow/lite/testing/op_tests/control_dep.py b/tensorflow/lite/testing/op_tests/control_dep.py
index b720fe5..f6467e5 100644
--- a/tensorflow/lite/testing/op_tests/control_dep.py
+++ b/tensorflow/lite/testing/op_tests/control_dep.py
@@ -38,7 +38,8 @@
input_tensor = tf.compat.v1.placeholder(
dtype=tf.float32, name="input", shape=parameters["input_shape"])
filter_value = tf.zeros((3, 3, TEST_INPUT_DEPTH, 8), tf.float32)
- assert_op = tf.assert_greater_equal(input_tensor, input_tensor - 1)
+ assert_op = tf.compat.v1.assert_greater_equal(input_tensor,
+ input_tensor - 1)
with tf.control_dependencies([assert_op]):
out = tf.nn.conv2d(
input_tensor, filter_value, strides=(1, 1, 1, 1), padding="SAME")