Remove duplicate tests
PiperOrigin-RevId: 350159389
Change-Id: I6dcff0f8ceaf5bad4f3be67af7245d555a1f04bc
diff --git a/tensorflow/python/eager/def_function_test.py b/tensorflow/python/eager/def_function_test.py
index 2b1dad4..bc41a07 100644
--- a/tensorflow/python/eager/def_function_test.py
+++ b/tensorflow/python/eager/def_function_test.py
@@ -1000,38 +1000,6 @@
self.assertAllEqual(obj2.testDouble.experimental_get_tracing_count(), 3)
self.assertAllEqual(obj1.testDouble.experimental_get_tracing_count(), 2)
- def test_experimental_get_tracing_count_function(self):
-
- @def_function.function
- def double(a):
- return a + a
-
- double(constant_op.constant(1))
- double(constant_op.constant(2))
- self.assertAllEqual(double.experimental_get_tracing_count(), 1)
- double(constant_op.constant('a'))
- self.assertAllEqual(double.experimental_get_tracing_count(), 2)
-
- def test_experimental_get_tracing_count_method(self):
-
- class TestClass():
-
- @def_function.function
- def testDouble(self, a):
- return a + a
-
- obj1 = TestClass()
- obj1.testDouble(constant_op.constant(1))
- obj1.testDouble(constant_op.constant(2))
- obj1.testDouble(constant_op.constant(1.1))
- self.assertAllEqual(obj1.testDouble.experimental_get_tracing_count(), 2)
- obj2 = TestClass()
- obj2.testDouble(constant_op.constant(1))
- obj2.testDouble(constant_op.constant(1.1))
- obj2.testDouble(constant_op.constant('a'))
- self.assertAllEqual(obj2.testDouble.experimental_get_tracing_count(), 3)
- self.assertAllEqual(obj1.testDouble.experimental_get_tracing_count(), 2)
-
if __name__ == '__main__':
ops.enable_eager_execution()
