• Y
    add masked_select_grad kernel (#48137) · db0ea0ce
    ykkk2333 提交于
    * add stat tool
    
    * add roll and roll_grad kernels and strided_slice and strided_slice_grad kernels, test=kunlun
    
    * add masked_selected_grad kernel,test=kunlun
    db0ea0ce
sgd_kernel.cc 5.5 KB