| 419 | } |
| 420 | |
| 421 | TfLiteStatus Subgraph::SetOutputs(std::vector<int> outputs) { |
| 422 | TF_LITE_ENSURE_OK( |
| 423 | &context_, CheckTensorIndices("outputs", outputs.data(), outputs.size())); |
| 424 | outputs_ = std::move(outputs); |
| 425 | return kTfLiteOk; |
| 426 | } |
| 427 | |
| 428 | TfLiteStatus Subgraph::SetVariables(std::vector<int> variables) { |
| 429 | TF_LITE_ENSURE_OK(&context_, CheckTensorIndices("variables", variables.data(), |