@function.defun
def scan():
    return functional_ops.scan((lambda a, x: (a + x)), elems, parallel_iterations=1)
@function.defun
def scan():
    return functional_ops.scan((lambda a, x: (a + x)), elems, parallel_iterations=1)