commit f27bc524ddc9d3082e08d93f66788770fd674fc0
author Gaurav Jain <gjn@google.com> Thu May 28 15:18:31 2020 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Thu May 28 15:28:30 2020 -0700
tree 0de8e88adcfa440d0875ac3192648daba86b3b51
parent 4e4a4496b7f89b7a3b56bf48257a5f6c560f8dd3

Add benchmark for no-op convert of an EagerTensor

PiperOrigin-RevId: 313669416
Change-Id: I82fe1ad6bda97a6a69c8f0d69192d762d53d2034

tensorflow/python/eager/benchmarks_test.py

diff --git a/tensorflow/python/eager/benchmarks_test.py b/tensorflow/python/eager/benchmarks_test.py
index 72eaa66..5d57f1d 100644
--- a/tensorflow/python/eager/benchmarks_test.py
+++ b/tensorflow/python/eager/benchmarks_test.py
@@ -1231,6 +1231,14 @@
 
     self._run(fn, 10000)
 
+  def benchmark_convert_tensor(self):
+    value = ops.convert_to_tensor(42)
+
+    def fn():
+      return ops.convert_to_tensor(value)
+
+    self._run(fn, 10000)
+
   def _benchmark_convert_constant(self, value, cached):
     global GLOBAL_TEST_VALUE
     GLOBAL_TEST_VALUE = value