Pass the correct operands for branch computations when invoking EinsumHeightAnalysis::HandleCalledComputation.
Pass the correct operands for branch computations when invoking EinsumHeightAnalysis::HandleCalledComputation.

Reverts e5c11f3d1e1bef39cb1ea0937ac5addf04942f32
