def f(a, b): return array_ops.sum(math_ops.sqrt(math_ops.exp(a)) + b)
def f(a, b): y = array_ops.sum(math_ops.sqrt(math_ops.exp(a)) + b) if has_aux: return y, array_ops.asarray(1) else: return y