- 09 3月, 2022 1 次提交
-
-
由 zhangkaihuo 提交于
Sparse conv3d backward(gpu)
-
- 03 3月, 2022 1 次提交
-
-
由 zyfncg 提交于
* suppport sparse api in yaml * support auto-gen code of sparse api * do some refactor * add unittest test_sparse_conv_api * add unitest file Co-authored-by: Nzkh2016 <zhangkaihuo@baidu.com>
-
- 02 3月, 2022 1 次提交
-
-
由 zhangkaihuo 提交于
Sparse Conv3d backward code
-
- 28 2月, 2022 1 次提交
-
-
由 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
-