| 148 | } |
| 149 | |
| 150 | Tensor *INode::input(size_t idx) const |
| 151 | { |
| 152 | ARM_COMPUTE_ERROR_ON(_graph == nullptr); |
| 153 | ARM_COMPUTE_ERROR_ON(idx >= _input_edges.size()); |
| 154 | Edge *e = _graph->edge(_input_edges[idx]); |
| 155 | return (e != nullptr) ? e->tensor() : nullptr; |
| 156 | } |
| 157 | |
| 158 | Tensor *INode::output(size_t idx) const |
| 159 | { |
no test coverage detected