experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add benchmark for no-op convert of an EagerTensor

Browse Source 
PiperOrigin-RevId: 313669416
Change-Id: I82fe1ad6bda97a6a69c8f0d69192d762d53d2034
This commit is contained in:
Gaurav Jain 2020-05-28 15:18:31 -07:00 committed by TensorFlower Gardener
parent 4e4a4496b7
commit f27bc524dd
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/python/eager/benchmarks_test.py
View File

@ -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