1. 19 1月, 2023 1 次提交
    • J
      [KUNLUN] add op: maxpool_with_index (#49505) · f71f77e9
      jameszhang 提交于
      * [KUNLUN] add op: maxpool_with_index
      
      * use DeviceContext::Alloc() instead of DenseTensor::mutable_data()
      
      * fix file format
      
      * solve clip unittest failure
      
      * minor fix
      
      * Revert "solve clip unittest failure" since the issue is fixed
      in #49535
      
      This reverts commit 1127adc66e79afe35ac3c00bb34e6aaa7cd7d78b.
      
      * align with xdnn on the definition of mask in max_pool_with_index
      
      * minor
      f71f77e9
  2. 18 1月, 2023 11 次提交
  3. 17 1月, 2023 17 次提交
  4. 16 1月, 2023 10 次提交
  5. 15 1月, 2023 1 次提交
    • J
      【Prim】Enhance tests (#49814) · 090aa45d
      Jiabin Yang 提交于
      * support elementwise base func
      
      * fix compiling error and add test
      
      * remove additional param
      
      * support vjp for div using comp
      
      * remove additional change
      
      * fix dy2st error with magic num
      
      * fix dy magic num
      
      * another magic
      
      * another magic
      
      * add more test
      
      * fix windows problem
      
      * another magic
      
      * fix windows compile
      
      * invoke ci
      
      * add skip rename strategy
      
      * support add vjp
      
      * fix test_tanh
      
      * support add with new axis cal
      
      * fix resnet and some test
      
      * add composite log
      
      * support sub vjp
      
      * enhance_tests
      
      * support more dtype for full
      090aa45d