1. 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
  2. 21 5月, 2019 1 次提交
  3. 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 696cf626.
      
      * Revert "change the output into inplace. test=develop"
      
      This reverts commit a19acd20.
      
      * test=develop.
      
      * update the MovingAverageAbsMaxScaleOp test. test=develop
      a914d9b1
  4. 21 3月, 2019 1 次提交
  5. 19 3月, 2019 1 次提交
  6. 15 3月, 2019 1 次提交
  7. 04 3月, 2019 1 次提交
  8. 03 9月, 2018 1 次提交
  9. 30 8月, 2018 1 次提交
  10. 28 8月, 2018 1 次提交
  11. 24 8月, 2018 1 次提交
  12. 11 7月, 2018 1 次提交