• X
    [ 0d-Tensor ] einsum support 0d tensor. (#49177) · 71bde066
    xiongkun 提交于
    * einsum support 0d tensor.
    1. support 0d tensor in multi-operands.
    2. add 9 unittests for einsum 0d tensor.
    
    * override NVIDIA_TF32_OVERRIDE to avoid accuracy problem in 11.2 and 11.8
    71bde066
einsum.py 35.4 KB