1. 05 1月, 2022 2 次提交
    • L
      optimize elementwise_mul_grad using new interfaces (#37728) · 36a102f8
      Lijunhui 提交于
      * init commit: new elem_mul_grad
      
      * add template speciallization for complex in multiply
      
      * reply review comments
      
      * correct dx and dy computation when T is complex
      
      * reply review comments
      
      * update to new ReduceRunctor
      
      * mul-output broadcast
      
      * call functions
      
      * call functions with comments
      
      * remove comments
      36a102f8
    • C
      implementation of broadcast div backward by reduce (#38044) · 55cd9cb8
      crystal 提交于
      * add elementwise div
      
      * move mul and div grad functor
      
      * Combine multiple CUDA kernels
      
      * Update the reduce interface call
      
      * add multi-output
      
      * add multi-output div
      
      * add branch judge
      
      * Package branch
      
      * Combine the x and y functions into one
      55cd9cb8
  2. 04 1月, 2022 1 次提交
  3. 16 12月, 2021 1 次提交
  4. 15 9月, 2021 1 次提交