From f0784e69761ef5b78480e9e8b1fd1aa558186646 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Tue, 9 Oct 2018 17:35:41 -0700 Subject: Add support for modeling fast memory close to the processor/gpu PiperOrigin-RevId: 216453979 --- tensorflow/core/grappler/costs/op_level_cost_estimator.h | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'tensorflow/core/grappler/costs/op_level_cost_estimator.h') 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 -- cgit v1.2.3