diff options
author | 2017-10-03 12:49:33 -0700 | |
---|---|---|
committer | 2017-10-03 12:53:46 -0700 | |
commit | 3f579020bab8f00e4621e9c7c740cbf13136a809 (patch) | |
tree | ea65a19dcbeebfa9889fd7a323ac7d3b08acef6a /tensorflow/python/profiler | |
parent | 9be96491599cd8890092f7010d4afd22862b26dd (diff) |
Convert cells to OO-based to reduce call() overhead
PiperOrigin-RevId: 170898081
Diffstat (limited to 'tensorflow/python/profiler')
-rw-r--r-- | tensorflow/python/profiler/internal/run_metadata_test.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/python/profiler/internal/run_metadata_test.py b/tensorflow/python/profiler/internal/run_metadata_test.py index c0de08cad6..80df44f5f5 100644 --- a/tensorflow/python/profiler/internal/run_metadata_test.py +++ b/tensorflow/python/profiler/internal/run_metadata_test.py @@ -140,7 +140,7 @@ class RunMetadataTest(test.TestCase): tfprof_node, run_meta = _run_loop_model() # The while-loop caused a node to appear 4 times in scheduling. ret = _extract_node(run_meta, - 'rnn/while/rnn/basic_rnn_cell/basic_rnn_cell/MatMul') + 'rnn/while/rnn/basic_rnn_cell/MatMul') self.assertEqual(len(ret['cpu:0']), 4) total_cpu_execs = 0 @@ -149,7 +149,7 @@ class RunMetadataTest(test.TestCase): mm_node = lib.SearchTFProfNode( tfprof_node, - 'rnn/while/rnn/basic_rnn_cell/basic_rnn_cell/MatMul') + 'rnn/while/rnn/basic_rnn_cell/MatMul') self.assertEqual(mm_node.run_count, 4) self.assertEqual(mm_node.cpu_exec_micros, total_cpu_execs) @@ -189,7 +189,7 @@ class RunMetadataTest(test.TestCase): tfprof_node, run_meta = _run_loop_model() # The while-loop caused a node to appear 4 times in scheduling. ret = _extract_node(run_meta, - 'rnn/while/rnn/basic_rnn_cell/basic_rnn_cell/MatMul') + 'rnn/while/rnn/basic_rnn_cell/MatMul') self.assertEqual(len(ret['gpu:0']), 4, '%s' % run_meta) total_cpu_execs = 0 |