diff options
author | Yao Zhang <yaozhang@google.com> | 2018-02-07 12:29:25 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-07 12:33:30 -0800 |
commit | 7ebc013313876748af71fde5af03859b9728181a (patch) | |
tree | 1a333ae89810f4e8fea2d40696aa66987f391d87 /tensorflow/python/grappler | |
parent | 65f676426d808cf20abd6a4a30ad48668ee5fdf2 (diff) |
Set the number of warmup steps for building the cost model.
PiperOrigin-RevId: 184878186
Diffstat (limited to 'tensorflow/python/grappler')
-rw-r--r-- | tensorflow/python/grappler/cluster.i | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/grappler/cluster.i b/tensorflow/python/grappler/cluster.i index 0c8d04ff29..8079cb307b 100644 --- a/tensorflow/python/grappler/cluster.i +++ b/tensorflow/python/grappler/cluster.i @@ -140,6 +140,7 @@ static GCluster TF_NewCluster(bool allow_soft_placement, timeout_s, num_cpu_cores, num_gpus); cluster_->DisableDetailedStats(disable_detailed_stats); cluster_->AllowSoftPlacement(allow_soft_placement); + cluster_->SetNumWarmupSteps(10); tensorflow::Status status = cluster_->Provision(); tensorflow::Set_TF_Status_from_Status(out_status, status); return GCluster(cluster_); |