There was an assumption that ops only have single tensor outputs. However ops like split return multiple tensors. This fixes that.
