diff options
author | 2017-11-01 19:31:24 -0700 | |
---|---|---|
committer | 2017-11-01 19:35:18 -0700 | |
commit | 16fa134cfb576bfa690d7006864e555dc42c6b62 (patch) | |
tree | 90a98df106d3c806113660d3d2366987a74741d2 /tensorflow/python/profiler | |
parent | 53a4fcbdbad571e659203733f6a07ba82651d40b (diff) |
Convert BasicRNNCell and GRUCell to proper layers.
PiperOrigin-RevId: 174272860
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 4ff09d3800..4c915ac79a 100644 --- a/tensorflow/python/profiler/internal/run_metadata_test.py +++ b/tensorflow/python/profiler/internal/run_metadata_test.py @@ -169,7 +169,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/MatMul') + 'rnn/while/basic_rnn_cell/MatMul') self.assertEqual(len(ret['cpu:0']), 4) total_cpu_execs = 0 @@ -178,7 +178,7 @@ class RunMetadataTest(test.TestCase): mm_node = lib.SearchTFProfNode( tfprof_node, - 'rnn/while/rnn/basic_rnn_cell/MatMul') + 'rnn/while/basic_rnn_cell/MatMul') self.assertEqual(mm_node.run_count, 4) self.assertEqual(mm_node.cpu_exec_micros, total_cpu_execs) @@ -218,7 +218,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/MatMul') + 'rnn/while/basic_rnn_cell/MatMul') self.assertEqual(len(ret['gpu:0']), 4, '%s' % run_meta) total_cpu_execs = 0 |