• Brandon T. Willard's avatar
    Stop inner-graph output reordering in Scan · 0b97f658
    Brandon T. Willard 提交于
    In order to use shared updates to pre-allocate the storage for mit-mot input and
    output loops, `Scan` would need to remove the corresponding mit-mot outputs from
    its inner-`FunctionGraph` before compilation and it would expect the `Function`
    compilation pipeline to add them back at the end of the remaining outputs.
    
    Now, `Scan`'s inner-`FunctionGraph`s maintain the same form at every point, and
    no special logic is needed to compensate for post-compilation changes in the
    order/location of inputs and outputs.
    0b97f658
名称
最后提交
最后更新
..
linalg 正在载入提交数据...
__init__.py 正在载入提交数据...
test_minimal.py 正在载入提交数据...
test_multinomial.py 正在载入提交数据...
test_multinomial_wo_replacement.py 正在载入提交数据...
test_rng_mrg.py 正在载入提交数据...