Documentation

mapWithIndex :: (Shape sh, Elt a, Elt b) => (Exp sh -> Exp a -> Exp b) -> Acc (Array sh a) -> Acc (Array sh b) Source #

gather :: (Shape ix, Shape ix', Elt ix', Elt a) => Acc (Array ix ix') -> Acc (Array ix' a) -> Acc (Array ix a) Source #

scatter :: (Shape ix, Shape ix', Elt ix', Elt a) => (Exp a -> Exp a -> Exp a) -> Acc (Array ix ix') -> Acc (Array ix' a) -> Acc (Array ix a) -> Acc (Array ix' a) Source #