1. 07 9月, 2021 5 次提交
  2. 06 9月, 2021 12 次提交
  3. 04 9月, 2021 1 次提交
  4. 03 9月, 2021 20 次提交
  5. 02 9月, 2021 2 次提交
    • J
      [NPU] Support npu kernel for gather_nd op (#34800) · bb633965
      JingZhuangzhuang 提交于
      * [NPU] Support npu kernel for gather_ng op
      
      * [NPU] Support npu kernel for gather_nd op
      
      * [NPU] Support npu kernel for gather_nd and gather_nd_grad op
      
      * update py format error.
      
      * modify gather_nd_op_npu
      
      * modify gather_nd 910 test
      
      * modify gather_nd 910 test
      Co-authored-by: Nxiaoxiaohehe001 <hiteezsf@163.com>
      bb633965
    • X
      Add SVD Op and it's GPU and CPU kernel (#34953) · 7e5fb462
      xiongkun 提交于
      * Add SVD Op and it's GPU and CPU kernel
      
      * Remove CUDAPlace in test_svd_op, make the test available in CPU package
      
      * modfity the file
      
      * fix windows bug/ fix ROCM / fix test timeout
      
      * for pass the CIs
      
      * improve error report
      
      * for code review
      
      * some modification to test_svd_op
      
      * change python code style
      
      * expose the svd interface for document
      7e5fb462