1. 02 3月, 2022 1 次提交
  2. 28 2月, 2022 1 次提交
    • Z
      Add sparse conv3d kernel (#39879) · bc99a76c
      zhangkaihuo 提交于
      * fix incorrect dims settings
      
      * sparse conv3d
      
      * fix out dims
      
      * test performance
      
      * test large shape success
      
      * opt scatter, double performance
      
      * test float16
      
      * remove profiling code
      
      * remove pten
      
      * opt code lines
      
      * correct boundary judgment
      
      * only cpu
      
      * test ci
      
      * test ci
      
      * remove the including paddle/fluid header; extract the conmmon function
      
      * opt code lines
      
      * use DenseTensor::data() instead of mutable_data
      
      * return rulebook for backward
      
      * specify layout
      
      * rename:conv -> sparse_conv3d
      bc99a76c