futhark-0.25.4: An optimising compiler for a functional, array-oriented language.
Safe HaskellSafe-Inferred
LanguageHaskell2010

Futhark.AD.Rev.Map

Description

VJP transformation for Map SOACs. This is a pretty complicated case due to the possibility of free variables.

Synopsis

Documentation

vjpMap :: VjpOps -> [Adj] -> StmAux () -> SubExp -> Lambda SOACS -> [VName] -> ADM () Source #

Perform VJP on a Map. The Adj list is the adjoints of the result of the map.