| 288 | } |
| 289 | |
| 290 | TfLiteStatus ArenaPlanner::CalculateAllocationOfInternalTensors( |
| 291 | int node_index) { |
| 292 | if (node_index < static_cast<int>(graph_info_->num_nodes())) { |
| 293 | const TfLiteNode& node = graph_info_->node(static_cast<size_t>(node_index)); |
| 294 | TfLiteIntArray* node_temporaries = node.temporaries; |
| 295 | for (int i = 0; i < node_temporaries->size; ++i) { |
| 296 | int tensor_index = node_temporaries->data[i]; |
| 297 | TF_LITE_ENSURE_STATUS(CalculateTensorAllocation(tensor_index)); |
| 298 | } |
| 299 | } |
| 300 | return kTfLiteOk; |
| 301 | } |
| 302 | |
| 303 | TfLiteStatus ArenaPlanner::CalculateDeallocationOfInternalTensors( |
| 304 | int node_index) { |