diff options
Diffstat (limited to 'tensorflow/tools/tfprof/internal/tfprof_timeline_test.cc')
| -rw-r--r-- | tensorflow/tools/tfprof/internal/tfprof_timeline_test.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/tools/tfprof/internal/tfprof_timeline_test.cc b/tensorflow/tools/tfprof/internal/tfprof_timeline_test.cc index 0e9bb9658c..cad31050a9 100644 --- a/tensorflow/tools/tfprof/internal/tfprof_timeline_test.cc +++ b/tensorflow/tools/tfprof/internal/tfprof_timeline_test.cc @@ -52,6 +52,7 @@ class TFProfTimelineTest : public ::testing::Test { tf_stats_.reset(new TFStats(std::move(graph_pb), std::move(run_meta_pb), nullptr, nullptr)); + tf_stats_->BuildAllViews(); } std::unique_ptr<TFStats> tf_stats_; |