| 409 | } |
| 410 | |
| 411 | Tensor* GraphExecutor::GetInputNodeTensor(unsigned int node_idx, unsigned int tensor_idx) |
| 412 | { |
| 413 | if(node_idx >= graph_->input_nodes.size()) |
| 414 | return nullptr; |
| 415 | |
| 416 | Node* node = graph_->input_nodes[node_idx]; |
| 417 | |
| 418 | if(tensor_idx >= node->GetOutputNum()) |
| 419 | return nullptr; |
| 420 | |
| 421 | return node->GetOutputTensor(tensor_idx); |
| 422 | } |
| 423 | |
| 424 | Tensor* GraphExecutor::GetOutputNodeTensor(unsigned int node_idx, unsigned int tensor_idx) |
| 425 | { |