aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/grappler/costs/op_level_cost_estimator.h
diff options
context:
space:
mode:
authorGravatar Benoit Steiner <bsteiner@google.com>2017-05-12 16:27:02 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-05-12 16:31:01 -0700
commite986f2d6965cd45ef5c90673518d259c028e3134 (patch)
treef30840d0db1bbbfdd00e63d841317f3d2aef1302 /tensorflow/core/grappler/costs/op_level_cost_estimator.h
parent0f34fa5b447c1bf22e0f509b95474c37aa5c98ba (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.h2
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_;
};