diff options
Diffstat (limited to 'tensorflow/tools/tfprof/internal/tfprof_tensor_test.cc')
-rw-r--r-- | tensorflow/tools/tfprof/internal/tfprof_tensor_test.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/tools/tfprof/internal/tfprof_tensor_test.cc b/tensorflow/tools/tfprof/internal/tfprof_tensor_test.cc index d7cfe7dade..0482657ca5 100644 --- a/tensorflow/tools/tfprof/internal/tfprof_tensor_test.cc +++ b/tensorflow/tools/tfprof/internal/tfprof_tensor_test.cc @@ -50,6 +50,7 @@ class TFProfTensorTest : public ::testing::Test { tf_stats_.reset(new TFStats(std::move(graph_pb), std::move(run_meta_pb), std::move(op_log_pb), std::move(ckpt_reader))); + tf_stats_->BuildAllViews(); } std::unique_ptr<TFStats> tf_stats_; |