From ee108441201ecb5fa9536573637623d712f9aa33 Mon Sep 17 00:00:00 2001 From: Max Galkin Date: Wed, 21 Mar 2018 12:53:04 -0700 Subject: Further improve accuracy of op_level_cost_estimator (Gather, GatherV2, Slice). PiperOrigin-RevId: 189952132 --- tensorflow/core/grappler/costs/op_level_cost_estimator.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'tensorflow/core/grappler/costs/op_level_cost_estimator.h') 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. -- cgit v1.2.3