| 412 | } |
| 413 | |
| 414 | TfLiteStatus Subgraph::SetInputs(std::vector<int> inputs) { |
| 415 | TF_LITE_ENSURE_OK(&context_, |
| 416 | CheckTensorIndices("inputs", inputs.data(), inputs.size())); |
| 417 | inputs_ = std::move(inputs); |
| 418 | return kTfLiteOk; |
| 419 | } |
| 420 | |
| 421 | TfLiteStatus Subgraph::SetOutputs(std::vector<int> outputs) { |
| 422 | TF_LITE_ENSURE_OK( |