1. 21 9月, 2020 1 次提交
    • H
      Quant op dev (#25932) · 02606d45
      huangxu96 提交于
      * Finished ChannelWiseQuantDequantAbsMaxOp and Passed unittests.
      
      * Finished channel-wise quantize strategy in imperative quantization.
      
      * Added Cuda code of ChannelWiseQuantDequantMaxAbsOP
      Add Cuda code of ChannelWiseQuantDequantMaxAbsOp
      
      * Add quant_axis for channel_wise quant.
      
      * fixed a bug in unnitests, which will not trigger axis = 1 case and cannot meet the coverage rate requirement.
      
      * Added some assert infomation and fixed some coding style mistakes.
      02606d45
  2. 19 8月, 2020 1 次提交
    • C
      [Quantization] Conv2d_transpose and mul support channnelwise quantization (#25639) · 3f816bc8
      cc 提交于
      * Conv2d_transpose and mul support channnelwise quantization, test=develop
      * Skip collecting out threshold for output tensor of which the type is not fp32 or fp64, test=develop
      * Fix error in test_user_defined_quantization, test=develop
      * Add depthwise_conv_bn_fuse, test=develop
      * Add conv_transpose_bn_fuse_pass for post_training_quant, test=develop
      3f816bc8
  3. 29 7月, 2020 1 次提交
  4. 09 7月, 2020 1 次提交
  5. 19 3月, 2020 1 次提交
  6. 11 9月, 2019 1 次提交
    • H
      Replace TemporaryAllocator by CUDADeviceContextAllocator (#18989) · 12542320
      Huihuang Zheng 提交于
      TemporaryAllocator is a singleton used for allocating memory for Cudnn. Since it is a singleton, we can delete it for better performance in memory.
      
      We replace TemporaryAllocator by CUDADeviceContextAllocator and CUDADeviceContextAllocation, which uses stream callback to delete the memory allocated for the stream to avoid singleton.
      
      Also added data_feed_proto to operator to fix CI in CPU compilation
      12542320
  7. 21 5月, 2019 1 次提交
  8. 07 5月, 2019 1 次提交
    • Z
      Quant output scale (#17215) · a914d9b1
      Zhen Wang 提交于
      * Add MovingAverageAbsMaxScale operator which is only used for calculating the quantization scale.
      
      * test=develop
      
      * change the output into inplace. test=develop
      
      * Revert "test=develop"
      
      This reverts commit 696cf62699ba1e1c98f61f7345ac7060010eb29a.
      
      * Revert "change the output into inplace. test=develop"
      
      This reverts commit a19acd20f07eee82622701a3015e6e9c073a5e0b.
      
      * test=develop.
      
      * update the MovingAverageAbsMaxScaleOp test. test=develop
      a914d9b1
  9. 21 3月, 2019 1 次提交
  10. 19 3月, 2019 1 次提交
  11. 15 3月, 2019 1 次提交
  12. 04 3月, 2019 1 次提交
  13. 03 9月, 2018 1 次提交
  14. 30 8月, 2018 1 次提交
  15. 28 8月, 2018 1 次提交
  16. 24 8月, 2018 1 次提交
  17. 11 7月, 2018 1 次提交