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 | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/core/grappler/costs/op_level_cost_estimator.h b/tensorflow/core/grappler/costs/op_level_cost_estimator.h index 36ef6a5c61..b4302dc9e1 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator.h +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.h @@ -82,10 +82,10 @@ class OpLevelCostEstimator { bool* found_unknown_shapes) const; int64 CountBatchMatMulOperations(const OpInfo& op_features, bool* found_unknown_shapes) const; - int64 CountConv2DBackPropInputOperations(const OpInfo& op_features, + int64 CountConv2DBackpropInputOperations(const OpInfo& op_features, ConvolutionDimensions* conv_info, bool* found_unknown_shapes) const; - int64 CountConv2DBackPropFilterOperations(const OpInfo& op_features, + int64 CountConv2DBackpropFilterOperations(const OpInfo& op_features, ConvolutionDimensions* conv_info, bool* found_unknown_shapes) const; @@ -124,8 +124,8 @@ class OpLevelCostEstimator { // device. Costs PredictConv2D(const OpInfo& op_features) const; Costs PredictCwiseOp(const OpInfo& op_features) const; - Costs PredictConv2DBackPropInput(const OpInfo& op_features) const; - Costs PredictConv2DBackPropFilter(const OpInfo& op_features) const; + Costs PredictConv2DBackpropInput(const OpInfo& op_features) const; + Costs PredictConv2DBackpropFilter(const OpInfo& op_features) const; Costs PredictMatMul(const OpInfo& op_features) const; Costs PredictNoOp(const OpInfo& op_features) const; Costs PredictBatchMatMul(const OpInfo& op_features) const; |