diff options
author | 2017-12-21 13:02:21 -0800 | |
---|---|---|
committer | 2017-12-21 13:06:08 -0800 | |
commit | 5b036664cbe829e92eb631af1e0f028f624df21e (patch) | |
tree | e28bbee071dea32fcb09c5526b0baa381e7d63e6 /tensorflow/contrib/lite/toco/allocate_transient_arrays.cc | |
parent | 77792130596305d9452079a1d3ddf3d186871741 (diff) |
Automated g4 rollback of changelist 179836027
PiperOrigin-RevId: 179848421
Diffstat (limited to 'tensorflow/contrib/lite/toco/allocate_transient_arrays.cc')
-rw-r--r-- | tensorflow/contrib/lite/toco/allocate_transient_arrays.cc | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/contrib/lite/toco/allocate_transient_arrays.cc b/tensorflow/contrib/lite/toco/allocate_transient_arrays.cc index d4da8f5dfe..62e7282d16 100644 --- a/tensorflow/contrib/lite/toco/allocate_transient_arrays.cc +++ b/tensorflow/contrib/lite/toco/allocate_transient_arrays.cc @@ -239,8 +239,8 @@ void AllocateTransientArrays(Model* model, // is a misnormer, should read 'workspace'. for (const auto& array_pair : ordered_arrays_map) { const string& array_name = array_pair.first; - auto it = array_lifespans.find(array_name); - if (it != array_lifespans.end() && it->second.persistent) { + const auto& array_lifespan = array_lifespans.find(array_name)->second; + if (array_lifespan.persistent) { AllocateTransientArray(*model, array_name, &allocator, transient_data_alignment); } @@ -282,8 +282,8 @@ void AllocateTransientArrays(Model* model, std::size_t persistent_alloc_size = 0; for (const auto& array_pair : ordered_arrays_map) { const string& array_name = array_pair.first; - auto it = array_lifespans.find(array_name); - if (it != array_lifespans.end() && it->second.persistent) { + const auto& array_lifespan = array_lifespans.find(array_name)->second; + if (array_lifespan.persistent) { persistent_alloc_size += TransientArraySize(*model, array_name, transient_data_alignment); } |