• Y
    [PTen]Refactor flatten_grad kernel (#38712) · 5cf0bb79
    YuanRisheng 提交于
    * refactor flatten grad kernel
    
    * fix bugs when run ci unittest
    
    * fix bugs when use default GetExpectedPtenKernelArgs
    
    * xshape sometimes is has null holder ,fix this bugs
    5cf0bb79
flatten_op.h 6.4 KB