Futhark.AD.Rev.Map

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

Documentation

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

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