1. 12 1月, 2022 1 次提交
  2. 11 1月, 2022 1 次提交
    • Z
      【PTen】Add dot and matmul grad kernel in pten (#38713) · be817719
      zyfncg 提交于
      * refactor matmul directory in pten
      
      * fix merge conflict
      
      * add dot_grad kernel
      
      * add dot_grad kernel in pten
      
      * add matmul_grad kernel
      
      * update the code
      
      * delete useless code in fluid
      
      * fix some bug of running matmul grad kernel
      
      * fix merge conflict
      
      * refactor some code
      
      * refactor code
      be817719
  3. 23 12月, 2021 1 次提交
  4. 21 12月, 2021 1 次提交
  5. 20 12月, 2021 1 次提交
  6. 31 12月, 2020 1 次提交
    • C
      complex gradient matmul (#29966) · e012930a
      chentianyu03 提交于
      * dot op support complex types
      
      * matmul support complex types
      
      * add test case
      
      * matmul broadcast gradient support complex
      
      * move conjFunctor to complex_functor.h
      e012930a
  7. 17 12月, 2020 1 次提交
    • C
      add conj op for complex types (#29527) · 71063b81
      chentianyu03 提交于
      * add conj op for complex types
      
      * add conj for complex types
      
      * add more test case
      
      * add conj_op test
      
      * modify conj api and impl
      
      * add complex type for fill_constant_op xpu
      
      * add setConstant for complex type
      
      * remove complex conj test file
      
      * user define grad for test_conj_op
      
      * add test case for static mode of conj api
      
      * modify conj doc
      
      * change input args name to x
      
      * remove useless codes
      
      * conj support real types
      
      * add conj test case for real number
      71063b81
反馈
建议
客服 返回
顶部