| 308 | } |
| 309 | |
| 310 | Status Node::input_node(int idx, Node** n) const { |
| 311 | const Edge* e; |
| 312 | TF_RETURN_IF_ERROR(input_edge(idx, &e)); |
| 313 | if (e == nullptr) { |
| 314 | *n = nullptr; |
| 315 | } else { |
| 316 | *n = e->src(); |
| 317 | } |
| 318 | return Status::OK(); |
| 319 | } |
| 320 | |
| 321 | Status Node::input_node(int idx, const Node** const_n) const { |
| 322 | Node* n; |