diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-10-09 17:35:41 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-09 17:40:04 -0700 |
commit | f0784e69761ef5b78480e9e8b1fd1aa558186646 (patch) | |
tree | f81c110d2c15b2643a24ace54b78ff00009e7691 /tensorflow/core/grappler/costs/op_level_cost_estimator.h | |
parent | eaebeb1d4d939fb9fd0b75e32a76151cb517bfb6 (diff) |
Add support for modeling fast memory close to the processor/gpu
PiperOrigin-RevId: 216453979
Diffstat (limited to 'tensorflow/core/grappler/costs/op_level_cost_estimator.h')
-rw-r--r-- | tensorflow/core/grappler/costs/op_level_cost_estimator.h | 3 |
1 files changed, 2 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 dd1ee39cb2..84dd9213f7 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator.h +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.h @@ -54,7 +54,8 @@ class OpLevelCostEstimator { // Naive cost estimate based on the given operations count and the given total // io size in bytes. Sizes of op_info inputs and outputs are not taken into // consideration. - Costs PredictOpCountBasedCost(double operations, double total_io_bytes, + Costs PredictOpCountBasedCost(double operations, double input_io_bytes, + double output_io_bytes, const OpInfo& op_info) const; // This family of routines counts the number of operations to perform the |