TODO(b/121264966): Subgraphs added after cancellation is set will not get the cancellation function added to their context.
| 260 | // TODO(b/121264966): Subgraphs added after cancellation is set will not get the |
| 261 | // cancellation function added to their context. |
| 262 | void Interpreter::SetCancellationFunction(void* data, |
| 263 | bool (*check_cancelled_func)(void*)) { |
| 264 | for (auto& subgraph : subgraphs_) { |
| 265 | subgraph->SetCancellationFunction(data, check_cancelled_func); |
| 266 | } |
| 267 | } |
| 268 | |
| 269 | TfLiteStatus Interpreter::ModifyGraphWithDelegate(TfLiteDelegate* delegate) { |
| 270 | for (auto& subgraph : subgraphs_) { |