1. 28 1月, 2022 1 次提交
    • H
      Move digamma to pten (#39240) · 848ae7dc
      hong 提交于
      * move digamma to pten; test=develop
      
      * fix mutable_data bugs; test=develop
      
      * remove useless code; test=develop
      
      * remove kernel compute; test=develop
      
      * fix bug; test=develop
      848ae7dc
  2. 15 6月, 2021 1 次提交
    • Z
      Add digamma_op and unittest (#33278) · 02a6d49a
      zyfncg 提交于
      * Add digamma_op and unittest
      
      * add digamma_op api
      
      * remove special DigammaCudaKernel and correct some docs
      
      * remove unused headers
      
      * fix api doc error
      02a6d49a
  3. 28 5月, 2021 1 次提交
    • L
      Add lgamma_op kernel and unittest (#32913) · 2d3cbb49
      levi131 提交于
      * run pre-commit
      
      * use HOST or DEVICE instead of HOSTDEVICE in implementation of lgamma op
      
      * add test for fp32
      
      * add lgamma to op_threshold_white_list
      
      * add cuda kernel for lgamma kernel
      
      * modify numeric grad delta
      
      * fix small English issue
      
      * change LaunchElementwiseCudaKernel to LaunchSameDimsElementwiseCudaKernel
      2d3cbb49
  4. 20 1月, 2021 1 次提交
    • C
      make abs op support complex types (#30375) · 358106fc
      chentianyu03 提交于
      * rewrite abs op
      
      * rewrite abs op and remove abs in activation
      
      * remove abs register in old codes
      
      * fix abs_grad type error
      
      * fix abs double_grad output name error
      
      * modify abs_grad, abs_grad_grad functor for windows building
      
      * format code style
      
      * fix the bug of result is nan when the divisor is zero
      
      * add missing abs attr and add abs for float16
      358106fc