diff options
author | Benoit Steiner <bsteiner@google.com> | 2017-12-15 10:31:27 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-15 10:34:41 -0800 |
commit | 5582df1a70c00e34c1254f798585cfd3ca05d90c (patch) | |
tree | 9637a04e9ec35158388ab853a30094a01ddbb879 /tensorflow/core/grappler | |
parent | 6c01a35b39d030e3e96b1a961163a0b29e377639 (diff) |
Added a python API to estimate the performance of HW devices
PiperOrigin-RevId: 179210330
Diffstat (limited to 'tensorflow/core/grappler')
-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 c6f23ee0aa..5f541ccf04 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator.h +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.h @@ -35,7 +35,6 @@ class OpLevelCostEstimator { virtual Costs PredictCosts(const OpContext& op_context) const; - protected: // Basic device performance info, sufficient for roofline estimate. struct DeviceInfo { double gigaops; // Billions of operations executed per second. @@ -45,6 +44,7 @@ class OpLevelCostEstimator { // Returns basic device performance info. virtual DeviceInfo GetDeviceInfo(const DeviceProperties& device) const; + protected: // Predict cost of an op for which no accurate estimator is defined. Costs PredictCostOfAnUnknownOp(const OpContext& op_context) const; |