1. 11 2月, 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. 25 5月, 2021 1 次提交
    • C
      modify Ops to complex template (#33041) · 5fa44c34
      chentianyu03 提交于
      * modify conj, real, imag OP to complex template
      
      * replace with complex template to dot Op
      
      * replace with complex template to Abs Op
      
      * add support for complex64 and complex128
      5fa44c34
  4. 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
  5. 12 4月, 2020 1 次提交