• N
    Replace ReduceAmax/Amax.part.cu with KP (#43202) (#43263) · e161979e
    niuliling123 提交于
    Reduce amax/amin frobenius_norm_kerne原始实现为Eigen实现,文件编译时间较长,因此本PR将其替换为KP实现
    删除DefaultElementwiseOperator中重复功能支持,减少elementwise_double_grad OP编译时间
    e161979e
frobenius_norm_kernel.cu 1.6 KB