• Ricardo Vieira's avatar
    Apply useless blockwise rewrite when there are only dummy batch dims · 10c36d2a
    Ricardo Vieira 提交于
    Also extend eager rewrite to more Ops
    
    The Blockwise MatrixInverse grad test became more sensitive in float32, because desired stabilization rewrites (mainly `inv_as_solve`) that target Dot of Blockwise{MatrixInverse} are now triggered in the default blockwise grad but not in the non-default non-blockwise grad
    10c36d2a
test_blockwise.py 13.0 KB