• Ricardo Vieira's avatar
    Fix local_fill_sink rewrite for multiple output Elemwise Ops · 8c157a25
    Ricardo Vieira 提交于
    The changes get rid of the eager sink at the local node rewriter level. This was actually not working because the nested replacements referenced variables that were never part of the original fgraph and those replacements were being ignored altogether. Instead we wrap the rewrite in an in2out that will safely achieve the intended behavior.
    8c157a25
test_basic.py 66.1 KB