| 889 | } |
| 890 | |
| 891 | TfLiteStatus Subgraph::AddTensors(int tensors_to_add, |
| 892 | int* first_new_tensor_index) { |
| 893 | const size_t base_index = tensors_.size(); |
| 894 | if (first_new_tensor_index) *first_new_tensor_index = base_index; |
| 895 | tensors_.resize(tensors_.size() + tensors_to_add); |
| 896 | for (size_t i = base_index; i < tensors_.size(); i++) { |
| 897 | memset(&tensors_[i], 0, sizeof(tensors_[i])); |
| 898 | tensors_[i].buffer_handle = kTfLiteNullBufferHandle; |
| 899 | } |
| 900 | context_.tensors = tensors_.data(); |
| 901 | context_.tensors_size = tensors_.size(); |
| 902 | return kTfLiteOk; |
| 903 | } |
| 904 | |
| 905 | TfLiteStatus Subgraph::AddTensors(TfLiteContext* context, int tensors_to_add, |
| 906 | int* first_new_tensor_index) { |