diff options
author | Yuefeng Zhou <yuefengz@google.com> | 2017-06-05 16:32:22 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-06-05 16:35:57 -0700 |
commit | 0cc851c08f9ff4c084c98e28e72d3f99a78d161f (patch) | |
tree | ad35cbbd7e8beb946ae4994fdbaa2fa376f77edf /tensorflow/core/grappler/costs/cost_estimator.h | |
parent | 7d76a90be4902698519a086920914469692847a3 (diff) |
Call maxcut algorithm in the model_based_cost_estimator.
PiperOrigin-RevId: 158078511
Diffstat (limited to 'tensorflow/core/grappler/costs/cost_estimator.h')
-rw-r--r-- | tensorflow/core/grappler/costs/cost_estimator.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/costs/cost_estimator.h b/tensorflow/core/grappler/costs/cost_estimator.h index b3fb3522a3..758a29696d 100644 --- a/tensorflow/core/grappler/costs/cost_estimator.h +++ b/tensorflow/core/grappler/costs/cost_estimator.h @@ -17,6 +17,7 @@ limitations under the License. #define TENSORFLOW_GRAPPLER_COSTS_COST_ESTIMATOR_H_ #include <chrono> +#include <unordered_map> #include "tensorflow/core/lib/core/status.h" namespace tensorflow { @@ -94,6 +95,9 @@ struct Costs { // streams from main memory. // If the time estimation is inaccurate. bool inaccurate = false; + + // Max possible memory usage per device. + std::unordered_map<string, uint64> estimated_max_memory_per_device; }; inline std::ostream& operator<<(std::ostream& os, const Costs::MicroSeconds d) { |