diff options
Diffstat (limited to 'tensorflow/core/grappler/costs/op_level_cost_estimator.h')
| -rw-r--r-- | tensorflow/core/grappler/costs/op_level_cost_estimator.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/grappler/costs/op_level_cost_estimator.h b/tensorflow/core/grappler/costs/op_level_cost_estimator.h index c6f23ee0aa..5f541ccf04 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator.h +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.h @@ -35,7 +35,6 @@ class OpLevelCostEstimator { virtual Costs PredictCosts(const OpContext& op_context) const; - protected: // Basic device performance info, sufficient for roofline estimate. struct DeviceInfo { double gigaops; // Billions of operations executed per second. @@ -45,6 +44,7 @@ class OpLevelCostEstimator { // Returns basic device performance info. virtual DeviceInfo GetDeviceInfo(const DeviceProperties& device) const; + protected: // Predict cost of an op for which no accurate estimator is defined. Costs PredictCostOfAnUnknownOp(const OpContext& op_context) const; |