@dace.map
def compute_ty(i: _[0:N], j: _[0:N]):
    (ia << A[(i, j)])
    (ib << B[(i, j)])
    (ix << x[j])
    (ot >> tmp(1, (lambda a, b: (a + b)), 0)[i])
    (oy >> y(1, (lambda a, b: (a + b)), 0)[i])
    ot = (ia * ix)
    oy = (ib * ix)
@dace.map
def compute_ty(i: _[0:N], j: _[0:N]):
    (ia << A[(i, j)])
    (ib << B[(i, j)])
    (ix << x[j])
    (ot >> tmp(1, (lambda a, b: (a + b)), 0)[i])
    (oy >> y(1, (lambda a, b: (a + b)), 0)[i])
    ot = (ia * ix)
    oy = (ib * ix)