diff options
| author |  Yao Zhang <yaozhang@google.com> | 2017-11-21 14:47:00 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-21 14:50:39 -0800 |
| commit | f0d1abbf2389aa2a29fe6fd090ba68ab6b8fd76f (patch) | |
| tree | 2643563eb67ff3f773d241735f2ace003c62e371 /tensorflow/core/grappler/costs/op_level_cost_estimator.cc | |
| parent | c6117cacf6ca91e442aea9bacb88a9444be60b33 (diff) | |
Pass VirtualCluster to tf_optimizer to enable cross-optimization of models.
PiperOrigin-RevId: 176564391
Diffstat (limited to 'tensorflow/core/grappler/costs/op_level_cost_estimator.cc')
| -rw-r--r-- | tensorflow/core/grappler/costs/op_level_cost_estimator.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/core/grappler/costs/op_level_cost_estimator.cc b/tensorflow/core/grappler/costs/op_level_cost_estimator.cc index bd84331b67..f7905d7798 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator.cc +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.cc @@ -324,7 +324,8 @@ OpLevelCostEstimator::DeviceInfo OpLevelCostEstimator::GetDeviceInfo( // Maxwell cores_per_multiprocessor = 128; } else { - // Pascal + // Pascal (compute capability version 6) and Volta (compute capability + // version 7) cores_per_multiprocessor = 64; } gflops = device.num_cores() * device.frequency() * 1e-3 * |