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/core/grappler/costs/cost_estimator.h | |
| parent | 49643265c3f1f279a93bd8bc3a126e11e979bc44 (diff) | |
Consolidate device parameter arguments into a shared DeviceInfo struct
PiperOrigin-RevId: 216280197
Diffstat (limited to 'tensorflow/core/grappler/costs/cost_estimator.h')
| -rw-r--r-- | tensorflow/core/grappler/costs/cost_estimator.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/costs/cost_estimator.h b/tensorflow/core/grappler/costs/cost_estimator.h index e91f0cc9da..569d9da683 100644 --- a/tensorflow/core/grappler/costs/cost_estimator.h +++ b/tensorflow/core/grappler/costs/cost_estimator.h @@ -30,6 +30,11 @@ struct GrapplerItem; constexpr int64 kMemoryUnknown = -1ll; constexpr int64 kZeroMemory = 0ll; +struct DeviceInfo { + double gigaops; // Billions of operations executed per second. + double gb_per_sec; // Bandwidth to main memory in GB per second. +}; + // Holds the set of things we might want to estimate or measure in Grappler. // Always produce execution time. Other fields are optional depending on the // estimator being used. |