spconv:PyTorch中的空间稀疏卷积
需积分: 45 27 浏览量
2021-05-01
03:07:21
上传
评论
收藏 256KB ZIP 举报
spconv-master.zip （96个子文件） 
spconv-master .github workflows 
build.yaml 1KB .gitmodules 309B spconv identity.py 361B utils __init__.py 12KB pool.py 3KB ops.py 6KB tables.py 1KB __init__.py 4KB modules.py 7KB functional.py 5KB test_utils.py 8KB conv.py 19KB src utils 
CMakeLists.txt 1KB all.cc 6KB nms.cu 6KB spconv indice.cu 16KB nms.cc 5KB reordering.cu 15KB indice.cc 13KB cublas_gemm.cc 2KB CMakeLists.txt 803B all.cc 2KB nms.cu 3KB pool_ops.cc 3KB maxpool.cu 18KB spconv_ops.cc 24KB maxpool.cc 3KB reordering.cc 3KB pillar_scatter.cu 2KB cuhash main.cc 1KB hash_table.cu 5KB debugging.cpp 3KB hash_functions.cpp 351B hash_table.cpp 7KB debugging.cu 8KB CMakeLists.txt 1KB hash_functions.cu 1KB Dockerfile 648B PERFORMANCE_GUIDE.md 2KB CMakeLists.txt 2KB example mnist_sparse.py 6KB LICENSE 11KB test fake_train.py 5KB src catch_main.cpp 481B test_conv_rule.cpp 5KB CMakeLists.txt 1KB test_conv.py 30KB fake_dist_train.py 5KB setup.py 5KB README.md 6KB format_all.sh 271B .gitignore 1KB third_party catch2 catch.hpp 466KB mp11 cutlass pybind11 CHANGELOG.md 693B include tensorview eigen_utils.h 1KB tensorview.h 51KB tools.h 2KB common.h 5KB pybind_utils.h 5KB mp_helper.h 1KB cuda_utils.h 722B torch_utils.h 5KB tensor.h 31KB prettyprint.h 16KB kernel_utils.h 2KB tensorrt inference.h 7KB spconv fused_spconv_ops.h 5KB nms_ops.h 2KB nms_functor.h 1KB indice.cu.h 18KB reordering.cu.h 15KB geometry.h 5KB pool_ops.h 1KB point2voxel.h 10KB pillar_scatter_ops.h 2KB maxpool.h 2KB cublas_gemm.h 2KB box_iou.h 6KB nms_gpu.h 781B pillar_scatter_functor.h 1KB spconv_ops.h 3KB reordering.h 2KB indice.h 2KB nms.h 7KB cuhash hash_table.h 9KB hash_table.cuh 9KB cuda_util.h 3KB debugging.h 3KB definitions.h 4KB hash_functions.h 3KB utility timer.h 2KB torch_utils.h 4KB paramsgrid.h 2KB tsl robin_map.h 24KB robin_growth_policy.h 11KB robin_hash.h 44KB资源评论
LiuTitanium
- 粉丝: 23
- 资源: 4684
