| 268 | } |
| 269 | |
| 270 | NodeDef* GetInputNode(const NodeDef& node, const MutableGraphView& graph) { |
| 271 | if (node.input_size() == 0) return nullptr; |
| 272 | MutableGraphView::InputPort input_port = graph.GetInputPort(node.name(), 0); |
| 273 | return graph.GetRegularFanin(input_port).node; |
| 274 | } |
| 275 | |
| 276 | NodeDef* GetInputNode(const NodeDef& node, const MutableGraphView& graph, |
| 277 | int64 i) { |