diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-10-08 17:33:22 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-08 17:38:39 -0700 |
| commit | 03d097bc96080981098ffdbaf1b3465e6e153a6a (patch) | |
| tree | 3ab671663167deda026298a6e85f09376c4b5d22 /tensorflow/core/grappler/costs/op_level_cost_estimator.cc | |
| parent | 49643265c3f1f279a93bd8bc3a126e11e979bc44 (diff) | |
Consolidate device parameter arguments into a shared DeviceInfo struct
PiperOrigin-RevId: 216280197
Diffstat (limited to 'tensorflow/core/grappler/costs/op_level_cost_estimator.cc')
| -rw-r--r-- | tensorflow/core/grappler/costs/op_level_cost_estimator.cc | 2 |
1 files changed, 1 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 71f4d9fd05..f363f2915f 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator.cc +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.cc @@ -372,7 +372,7 @@ Costs OpLevelCostEstimator::PredictCosts(const OpContext& op_context) const { return costs; } -OpLevelCostEstimator::DeviceInfo OpLevelCostEstimator::GetDeviceInfo( +DeviceInfo OpLevelCostEstimator::GetDeviceInfo( const DeviceProperties& device) const { double gflops = -1; double gb_per_sec = -1; |