experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add const to helper methods in grappler::NodeCosts.

Browse Source 
PiperOrigin-RevId: 347648657
Change-Id: I388b5a971837fb8014cfe17796dcc75c2b057db1
This commit is contained in:
Doe Hyun Yoon 2020-12-15 10:57:07 -08:00 committed by TensorFlower Gardener
parent a8179619ee
commit ba9e8f7891
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/core/grappler/costs/op_level_cost_estimator.h
View File

@ -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();
}