| 408 | } |
| 409 | |
| 410 | Tensor* Session::getInput(const char* name) const { |
| 411 | // MNN_ASSERT(!mInputs.empty()); |
| 412 | if (nullptr == name) { |
| 413 | return mInfo.inputTensors.begin()->second; |
| 414 | } |
| 415 | auto iter = mInfo.inputTensors.find(name); |
| 416 | if (iter == mInfo.inputTensors.end()) { |
| 417 | MNN_PRINT("Error: can't find input: %s\n", name); |
| 418 | return nullptr; |
| 419 | } |
| 420 | return iter->second; |
| 421 | } |
| 422 | Tensor* Session::getTensor(int index) const { |
| 423 | return mInfo.allTensors[index].get(); |
| 424 | } |