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

Merge pull request #39927 from tg-at-google:patch-5 

PiperOrigin-RevId: 314440108
Change-Id: Ib742967d9324da908f7a4102cdc015140176730c
This commit is contained in:
TensorFlower Gardener 2020-06-02 17:59:04 -07:00
parent a8b9c83afe 85dcf96a8a
commit 5656e1cb45
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/lite/arena_planner.cc
View File

@ -197,7 +197,8 @@ TfLiteStatus ArenaPlanner::ExecuteAllocations(int first_node, int last_node) {
dealloc_node_.resize(graph_info_->num_tensors(), kNodeNotAssigned); dealloc_node_.resize(graph_info_->num_tensors(), kNodeNotAssigned);
allocs_.resize(graph_info_->num_tensors()); allocs_.resize(graph_info_->num_tensors());
// Set allocation and deallocation for temporary tensors. // Set allocation and deallocation for temporary tensors.
for (size_t i = first_node; i <= last_node && i < graph_info_->num_nodes(); for (size_t i = first_node;
i <= static_cast<size_t>(last_node) && i < graph_info_->num_nodes();
++i) { ++i) {
const TfLiteNode& node = graph_info_->node(i); const TfLiteNode& node = graph_info_->node(i);
TfLiteIntArray* node_temporaries = node.temporaries; TfLiteIntArray* node_temporaries = node.temporaries;