aeppl.rewriting.PreserveRVMappings.on_change_input#

PreserveRVMappings.on_change_input(fgraph, node, i, r, new_r, reason=None)[source]#

Whenever a node is replaced during rewrite, we check if it had a value variable associated with it and map it to the new node.