• Y
    [Zero-Dim] Add Expand/Expand_as/Top_k for XPU to support Zero Dim Input. (#50947) · 226b4a95
    yunyaoXYY 提交于
    * Add unitest from shilong
    
    * Add kernel code from shilong
    
    * fix codestyle
    
    * add broadcast_shape test
    
    * fix unitest
    
    * fix unitests
    
    * fix unitest
    
    * add 0D grad support
    
    * add 0D grad support
    
    * add 0D grad support
    
    * fix 0D tensor
    
    * fix 0D
    
    * fix xpu 0D
    
    * fix expand kernel
    
    * fix xpu expand
    
    * Fix 0D kernel
    
    * fix 0D
    
    * fix 0D
    
    * fix 0D
    
    * fix 0D
    
    * fix XPU top_k
    
    * cancel the modify of xpu
    
    * add XPU 0D tensor
    
    * fix 0D
    226b4a95
expand_kernel.cc 5.3 KB