| 301 | } |
| 302 | |
| 303 | TfLiteStatus Interpreter::GetBufferHandle(int tensor_index, |
| 304 | TfLiteBufferHandle* buffer_handle, |
| 305 | TfLiteDelegate** delegate) { |
| 306 | TF_LITE_ENSURE(context_, tensor_index < tensors_size()); |
| 307 | std::vector<TfLiteTensor>& tensors = primary_subgraph().tensors(); |
| 308 | TfLiteTensor* tensor = &tensors[tensor_index]; |
| 309 | |
| 310 | *delegate = tensor->delegate; |
| 311 | *buffer_handle = tensor->buffer_handle; |
| 312 | |
| 313 | return kTfLiteOk; |
| 314 | } |
| 315 | |
| 316 | void Interpreter::SetProfiler(Profiler* profiler) { |
| 317 | for (auto& subgraph : subgraphs_) subgraph->SetProfiler(profiler); |