From f27bc524ddc9d3082e08d93f66788770fd674fc0 Mon Sep 17 00:00:00 2001 From: Gaurav Jain Date: Thu, 28 May 2020 15:18:31 -0700 Subject: [PATCH] Add benchmark for no-op convert of an EagerTensor PiperOrigin-RevId: 313669416 Change-Id: I82fe1ad6bda97a6a69c8f0d69192d762d53d2034 --- tensorflow/python/eager/benchmarks_test.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/tensorflow/python/eager/benchmarks_test.py b/tensorflow/python/eager/benchmarks_test.py index 72eaa663c6c..5d57f1d9b93 100644 --- a/tensorflow/python/eager/benchmarks_test.py +++ b/tensorflow/python/eager/benchmarks_test.py @@ -1231,6 +1231,14 @@ class MicroBenchmarks(benchmarks_test_base.MicroBenchmarksBase): 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