accelerate-utility-0.0: Utility functions for the Accelerate framework
Data.Array.Accelerate.Utility.Arrange
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