diff options
Diffstat (limited to 'tensorflow/python')
| -rw-r--r-- | tensorflow/python/grappler/cluster.i | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/grappler/cluster.i b/tensorflow/python/grappler/cluster.i index 6816e20407..87795ffcfb 100644 --- a/tensorflow/python/grappler/cluster.i +++ b/tensorflow/python/grappler/cluster.i @@ -308,7 +308,7 @@ static PyObject* TF_GetSupportedDevices(GCluster cluster, GItem item) { static double TF_EstimatePerformance(const tensorflow::NamedDevice& device) { tensorflow::grappler::OpLevelCostEstimator estimator; - tensorflow::grappler::OpLevelCostEstimator::DeviceInfo info = + tensorflow::grappler::DeviceInfo info = estimator.GetDeviceInfo(device.properties()); return info.gigaops; } |