diff options
author | 2017-05-12 16:27:02 -0700 | |
---|---|---|
committer | 2017-05-12 16:31:01 -0700 | |
commit | e986f2d6965cd45ef5c90673518d259c028e3134 (patch) | |
tree | f30840d0db1bbbfdd00e63d841317f3d2aef1302 /tensorflow/core/grappler/costs/op_level_cost_estimator.h | |
parent | 0f34fa5b447c1bf22e0f509b95474c37aa5c98ba (diff) |
Made the device_cost_impl_ accessible to subclasses.
PiperOrigin-RevId: 155925151
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 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<Costs(const OpInfo& op_feature)> CostImpl; std::map<string, CostImpl> device_cost_impl_; }; |