diff options
author | 2017-06-29 16:14:52 -0700 | |
---|---|---|
committer | 2017-06-29 16:19:27 -0700 | |
commit | 953a6db9b6c3473f40e6ba2db207c62ef0b19097 (patch) | |
tree | 6bf20f0a838f05d7833b13d88e32746cde0d0b2b /tensorflow/core/grappler/costs/op_level_cost_estimator.h | |
parent | 8280e0ae9083a65b23608b34723f07e028a56dc8 (diff) |
Improve the accuracy of the cost estimates for the size, shape, and rank ops.
PiperOrigin-RevId: 160587845
Diffstat (limited to 'tensorflow/core/grappler/costs/op_level_cost_estimator.h')
-rw-r--r-- | tensorflow/core/grappler/costs/op_level_cost_estimator.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/costs/op_level_cost_estimator.h b/tensorflow/core/grappler/costs/op_level_cost_estimator.h index 28d49a7703..59ced70ba6 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator.h +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.h @@ -129,6 +129,7 @@ class OpLevelCostEstimator { Costs PredictMatMul(const OpInfo& op_features) const; Costs PredictNoOp(const OpInfo& op_features) const; Costs PredictBatchMatMul(const OpInfo& op_features) const; + Costs PredictMetadata(const OpInfo& op_features) const; // Utility function for safe division. Returns 0 // if rhs is 0 or negative. |