| 422 | } |
| 423 | |
| 424 | Tensor* GraphExecutor::GetOutputNodeTensor(unsigned int node_idx, unsigned int tensor_idx) |
| 425 | { |
| 426 | if(node_idx >= graph_->output_nodes.size()) |
| 427 | return nullptr; |
| 428 | |
| 429 | Node* node = graph_->output_nodes[node_idx]; |
| 430 | |
| 431 | if(tensor_idx >= node->GetOutputNum()) |
| 432 | return nullptr; |
| 433 | |
| 434 | return node->GetOutputTensor(tensor_idx); |
| 435 | } |
| 436 | |
| 437 | bool GraphExecutor::SetTensorBuffer(Tensor* tensor, void* input_data, int data_size) |
| 438 | { |
nothing calls this directly
no test coverage detected