From 5582df1a70c00e34c1254f798585cfd3ca05d90c Mon Sep 17 00:00:00 2001 From: Benoit Steiner Date: Fri, 15 Dec 2017 10:31:27 -0800 Subject: Added a python API to estimate the performance of HW devices PiperOrigin-RevId: 179210330 --- 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 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; -- cgit v1.2.3