diff options
author | 2018-03-21 12:53:04 -0700 | |
---|---|---|
committer | 2018-03-21 12:55:38 -0700 | |
commit | ee108441201ecb5fa9536573637623d712f9aa33 (patch) | |
tree | 74f4313979bdd43f3292064f951ecd23345e0539 /tensorflow/core/grappler/costs/op_level_cost_estimator.h | |
parent | bdd6f2253a76c707ff2ce2af9b560478891342eb (diff) |
Further improve accuracy of op_level_cost_estimator (Gather, GatherV2, Slice).
PiperOrigin-RevId: 189952132
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 e5dd31a7a2..1b3babb206 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator.h +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.h @@ -144,7 +144,7 @@ class OpLevelCostEstimator { Costs PredictVariable(const OpContext& op_context) const; Costs PredictBatchMatMul(const OpContext& op_context) const; Costs PredictMetadata(const OpContext& op_context) const; - Costs PredictGather(const OpContext& op_context) const; + Costs PredictGatherOrSlice(const OpContext& op_context) const; // Utility function for safe division. Returns 0 // if rhs is 0 or negative. |