diff options
author | 2017-05-05 18:02:11 -0800 | |
---|---|---|
committer | 2017-05-05 19:25:59 -0700 | |
commit | 87ba9f5370c0f7068760f9536979d9183f6dfe9c (patch) | |
tree | da5bda113ca32c540a046546973ba7f2d50ff715 /tensorflow/core/grappler/costs/op_level_cost_estimator.h | |
parent | be2a9e5f35fdf34f1ff06a8cfdcb8ae38a6b50c9 (diff) |
Return more complete device information from the GetDevices() method of grappler clusters.
Change: 155264843
Diffstat (limited to 'tensorflow/core/grappler/costs/op_level_cost_estimator.h')
-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 5bb20cc6bb..7a594e2a01 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator.h +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.h @@ -40,7 +40,7 @@ class OpLevelCostEstimator { // executed per second) and memory bandwith (in GigaBytes/second) for the // specified device. virtual std::pair<double, double> GetDeviceInfo( - const OpInfo::DeviceProperties& device) const; + const DeviceProperties& device) const; // For operations for which we haven't yet built estimates, returns a dummy // value based on input size. |