| 169 | } |
| 170 | |
| 171 | void SingleOpModel::ApplyDelegate() { |
| 172 | if (force_use_nnapi) { |
| 173 | // TODO(b/124505407): Check the result and fail accordingly. |
| 174 | interpreter_->ModifyGraphWithDelegate(TestNnApiDelegate()); |
| 175 | } |
| 176 | |
| 177 | // Modify delegate with function. |
| 178 | if (apply_delegate_fn_) { |
| 179 | apply_delegate_fn_(interpreter_.get()); |
| 180 | } |
| 181 | } |
| 182 | |
| 183 | void SingleOpModel::Invoke() { ASSERT_EQ(interpreter_->Invoke(), kTfLiteOk); } |
| 184 |
no test coverage detected