From ff4dd5cf1c600ecceae36162cbfb3e219585ad4e Mon Sep 17 00:00:00 2001
From: Daryl Ng <darylng@google.com>
Date: Wed, 3 Jul 2019 08:27:29 -0700
Subject: [PATCH] Improving readability by using Cluster::NumWarmupSteps()
instead of directly calling
options_.config.graph_options().build_cost_model_after()
PiperOrigin-RevId: 256372140
---
tensorflow/core/grappler/clusters/single_machine.cc | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/tensorflow/core/grappler/clusters/single_machine.cc b/tensorflow/core/grappler/clusters/single_machine.cc
index f7d4cffdd3d..c44b74efcdc 100644
--- a/tensorflow/core/grappler/clusters/single_machine.cc
+++ b/tensorflow/core/grappler/clusters/single_machine.cc
@@ -181,8 +181,7 @@ Status SingleMachine::Run(const GraphDef& graph_def,
}
// Warmup TensorFlow if needed
- for (int i = 0;
- i < options_.config.graph_options().build_cost_model_after(); ++i) {
+ for (int i = 0; i < NumWarmupSteps(); ++i) {
TF_RETURN_IF_ERROR(RunWithTimeout(feed, fetch, nullptr));
}
}