diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-10-08 17:33:22 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-08 17:38:39 -0700 |
commit | 03d097bc96080981098ffdbaf1b3465e6e153a6a (patch) | |
tree | 3ab671663167deda026298a6e85f09376c4b5d22 /tensorflow/python | |
parent | 49643265c3f1f279a93bd8bc3a126e11e979bc44 (diff) |
Consolidate device parameter arguments into a shared DeviceInfo struct
PiperOrigin-RevId: 216280197
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; } |