Improving readability by using Cluster::NumWarmupSteps() instead of directly calling options_.config.graph_options().build_cost_model_after()

PiperOrigin-RevId: 256372140
2019-07-03 08:27:29 -07:00 · 2019-07-03 08:27:29 -07:00 · ff4dd5cf1c
commit ff4dd5cf1c
parent 9746ceaf48
1 changed files with 1 additions and 2 deletions
--- 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));
    }
  }