From ba9e8f7891d2001c0a8ff683abcf6ea2406fe504 Mon Sep 17 00:00:00 2001
From: Doe Hyun Yoon <dyoon@google.com>
Date: Tue, 15 Dec 2020 10:57:07 -0800
Subject: [PATCH] Add const to helper methods in grappler::NodeCosts.
PiperOrigin-RevId: 347648657
Change-Id: I388b5a971837fb8014cfe17796dcc75c2b057db1
---
.../core/grappler/costs/op_level_cost_estimator.h | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
diff --git a/tensorflow/core/grappler/costs/op_level_cost_estimator.h b/tensorflow/core/grappler/costs/op_level_cost_estimator.h
index 3d58d13f423..54382927f7b 100644
--- a/tensorflow/core/grappler/costs/op_level_cost_estimator.h
+++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.h
@@ -51,21 +51,21 @@ struct NodeCosts {
int64 internal_write_bytes = 0;
// Convenience functions.
- int64 num_total_input_bytes() {
+ int64 num_total_input_bytes() const {
return std::accumulate(num_input_bytes_accessed.begin(),
num_input_bytes_accessed.end(), 0LL);
}
- int64 num_total_read_bytes() {
+ int64 num_total_read_bytes() const {
return num_total_input_bytes() + internal_read_bytes;
}
- int64 num_total_output_bytes() {
+ int64 num_total_output_bytes() const {
return std::accumulate(num_output_bytes_accessed.begin(),
num_output_bytes_accessed.end(), 0LL);
}
- int64 num_total_write_bytes() {
+ int64 num_total_write_bytes() const {
return num_total_output_bytes() + internal_write_bytes;
}
- int64 num_bytes_accessed() {
+ int64 num_bytes_accessed() const {
return num_total_read_bytes() + num_total_write_bytes();
}