diff options
author | 2017-05-22 22:25:45 -0700 | |
---|---|---|
committer | 2017-05-22 22:29:28 -0700 | |
commit | ecaf99775d59b1905183833db9f5348b6bbd3f41 (patch) | |
tree | ffe66034f9d4d2c557a34cf0ea33ee8b7db7b777 /tensorflow/core/grappler/costs/op_level_cost_estimator.h | |
parent | e1b85cdd85471d7f6f19ce9eeeb4db616a5cc402 (diff) |
Made it possible to refine the behavior of the default op_level_cost_estimator.
PiperOrigin-RevId: 156827000
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 266b633922..5e0a97653c 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator.h +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.h @@ -32,7 +32,7 @@ class OpLevelCostEstimator { OpLevelCostEstimator(); virtual ~OpLevelCostEstimator() {} - Costs PredictCosts(const OpInfo& op_features) const; + virtual Costs PredictCosts(const OpInfo& op_features) const; protected: // Returns an estimate of device performance (in billions of operations |