From e986f2d6965cd45ef5c90673518d259c028e3134 Mon Sep 17 00:00:00 2001 From: Benoit Steiner Date: Fri, 12 May 2017 16:27:02 -0700 Subject: Made the device_cost_impl_ accessible to subclasses. PiperOrigin-RevId: 155925151 --- 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 78ce69a597..266b633922 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator.h +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.h @@ -132,7 +132,7 @@ class OpLevelCostEstimator { const TensorShapeProto& original_filter_shape, const OpInfo& op_features, bool* found_unknown_shapes); - private: + protected: typedef std::function CostImpl; std::map device_cost_impl_; }; -- cgit v1.2.3