Make Graph::UpdateEdge() be O(e) instead of O(E)

where:
- E = number of edges in the graph
- e = number of edges on the node of interest

e is necessarily <= E and is typically really small
(# of inputs to an operation + control edges)

PiperOrigin-RevId: 210624296