├── FLOPs_counter.py ├── LICENSE ├── README.md ├── SW.png ├── analysis ├── analysis.py ├── channelsparsity.pdf ├── ck_curve.py ├── cover.py ├── sparse_distribute.py ├── sparsity.pdf └── utilization.png ├── backbones ├── SW_v1_slak.py ├── SW_v2_unirep.py └── __init__.py ├── bmc_qr.png ├── curse.py ├── datasets.py ├── downstream ├── 3d_detection │ ├── SW_v2_unirep.py │ ├── configs │ │ ├── fcos3d.py │ │ └── fcos3d_UniRep_fpn_head-gn_8xb2-1x_nus-mono3d.py │ ├── readme.md │ └── unireplknet.py ├── README.md ├── detection │ ├── README.md │ ├── SW_v2_unirep.py │ ├── configs │ │ ├── casc_mask_rcnn_unireplknet_t_in1k_fpn_3x_coco.py │ │ ├── cascade_mask_rcnn_sw_fpn.py │ │ └── cascade_mask_rcnn_sw_tiny_mstrain_480-800_adamw_3x_coco_in1k.py │ ├── mmdet │ │ └── core │ │ │ └── optimizers │ │ │ ├── __init__.py │ │ │ └── layer_decay_optimizer_constructor.py │ ├── slak.py │ └── unireplknet.py └── segmentation │ ├── README.md │ ├── SW_v2_unirep.py │ ├── configs │ ├── upernet_SW.py │ ├── upernet_sw_tiny_512_160k_ade20k_ms.py │ ├── upernet_sw_tiny_512_160k_ade20k_ss.py │ └── upernet_unireplknet_s_512_160k_ade20k.py │ ├── mmseg │ └── core │ │ └── optimizers │ │ ├── __init__.py │ │ └── layer_decay_optimizer_constructor.py │ ├── slak.py │ └── unireplknet.py ├── engine.py ├── erf ├── SLaK_for_erf.py ├── SW_for_erf.py ├── Saved_matrix │ ├── ConvNext-T_31.npy │ ├── ConvNext-T_7.npy │ ├── SLaK-T_51.npy │ └── SW.npy ├── analyze_erf.py ├── erf_slak51_convnext7_convnext31.py ├── readme.md ├── resnet_for_erf.py ├── sw.pdf ├── sw_00.png └── visualize_erf.py ├── flopses.py ├── funcs.py ├── main.py ├── model_sema.py ├── models ├── SLaK.py ├── SLaK_pw_wt.py ├── SLaK_reg.py ├── __init__.py └── taichi_kernel.py ├── optim_factory.py ├── run_with_submitit.py ├── shiftadd ├── LICENSE ├── README.md ├── SLAK_se.py ├── SLaK_gr_sw_sf_se.py ├── SLaK_gr_sw_sf_se2.py ├── compare.py ├── compare_module.py ├── compare_mp_blur.py ├── compare_mp_embedding.py ├── compare_mp_linear.py ├── compare_mp_sp.py ├── is_jit_faster.py ├── ops │ ├── __init__.py │ ├── ops_py │ │ ├── __init__.py │ │ └── add_shift.py │ └── src │ │ ├── addshift_kernel_v0.cu │ │ ├── addshift_kernel_vmp.cu │ │ ├── addshift_kernel_vmp_blur.cu │ │ ├── addshift_kernel_vmp_linear.cu │ │ ├── addshift_kernel_vmp_pos_em.cu │ │ ├── addshift_ops_v0.cpp │ │ ├── addshift_ops_vmp.cpp │ │ ├── addshift_ops_vmp_blur.cpp │ │ ├── addshift_ops_vmp_linear.cpp │ │ └── addshift_ops_vmp_pos_em.cpp ├── setup.py └── test_ops.py ├── sparse_core.py ├── test └── depthwise_conv2d_test.py ├── throughout ├── a100fps.json ├── models │ ├── SLaK_gemm.py │ ├── SLaK_general.py │ ├── SW_v2_unirep.py │ ├── __init__.py │ ├── dw_gemm.py │ └── unireplknet.py ├── readme.md └── throughput.py └── utils.py /FLOPs_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/FLOPs_counter.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/README.md -------------------------------------------------------------------------------- /SW.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/SW.png -------------------------------------------------------------------------------- /analysis/analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/analysis/analysis.py -------------------------------------------------------------------------------- /analysis/channelsparsity.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/analysis/channelsparsity.pdf -------------------------------------------------------------------------------- /analysis/ck_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/analysis/ck_curve.py -------------------------------------------------------------------------------- /analysis/cover.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/analysis/cover.py -------------------------------------------------------------------------------- /analysis/sparse_distribute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/analysis/sparse_distribute.py -------------------------------------------------------------------------------- /analysis/sparsity.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/analysis/sparsity.pdf -------------------------------------------------------------------------------- /analysis/utilization.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/analysis/utilization.png -------------------------------------------------------------------------------- /backbones/SW_v1_slak.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/backbones/SW_v1_slak.py -------------------------------------------------------------------------------- /backbones/SW_v2_unirep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/backbones/SW_v2_unirep.py -------------------------------------------------------------------------------- /backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/backbones/__init__.py -------------------------------------------------------------------------------- /bmc_qr.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/bmc_qr.png -------------------------------------------------------------------------------- /curse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/curse.py -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/datasets.py -------------------------------------------------------------------------------- /downstream/3d_detection/SW_v2_unirep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/3d_detection/SW_v2_unirep.py -------------------------------------------------------------------------------- /downstream/3d_detection/configs/fcos3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/3d_detection/configs/fcos3d.py -------------------------------------------------------------------------------- /downstream/3d_detection/configs/fcos3d_UniRep_fpn_head-gn_8xb2-1x_nus-mono3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/3d_detection/configs/fcos3d_UniRep_fpn_head-gn_8xb2-1x_nus-mono3d.py -------------------------------------------------------------------------------- /downstream/3d_detection/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/3d_detection/readme.md -------------------------------------------------------------------------------- /downstream/3d_detection/unireplknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/3d_detection/unireplknet.py -------------------------------------------------------------------------------- /downstream/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/README.md -------------------------------------------------------------------------------- /downstream/detection/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/detection/README.md -------------------------------------------------------------------------------- /downstream/detection/SW_v2_unirep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/detection/SW_v2_unirep.py -------------------------------------------------------------------------------- /downstream/detection/configs/casc_mask_rcnn_unireplknet_t_in1k_fpn_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/detection/configs/casc_mask_rcnn_unireplknet_t_in1k_fpn_3x_coco.py -------------------------------------------------------------------------------- /downstream/detection/configs/cascade_mask_rcnn_sw_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/detection/configs/cascade_mask_rcnn_sw_fpn.py -------------------------------------------------------------------------------- /downstream/detection/configs/cascade_mask_rcnn_sw_tiny_mstrain_480-800_adamw_3x_coco_in1k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/detection/configs/cascade_mask_rcnn_sw_tiny_mstrain_480-800_adamw_3x_coco_in1k.py -------------------------------------------------------------------------------- /downstream/detection/mmdet/core/optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/detection/mmdet/core/optimizers/__init__.py -------------------------------------------------------------------------------- /downstream/detection/mmdet/core/optimizers/layer_decay_optimizer_constructor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/detection/mmdet/core/optimizers/layer_decay_optimizer_constructor.py -------------------------------------------------------------------------------- /downstream/detection/slak.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/detection/slak.py -------------------------------------------------------------------------------- /downstream/detection/unireplknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/detection/unireplknet.py -------------------------------------------------------------------------------- /downstream/segmentation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/segmentation/README.md -------------------------------------------------------------------------------- /downstream/segmentation/SW_v2_unirep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/segmentation/SW_v2_unirep.py -------------------------------------------------------------------------------- /downstream/segmentation/configs/upernet_SW.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/segmentation/configs/upernet_SW.py -------------------------------------------------------------------------------- /downstream/segmentation/configs/upernet_sw_tiny_512_160k_ade20k_ms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/segmentation/configs/upernet_sw_tiny_512_160k_ade20k_ms.py -------------------------------------------------------------------------------- /downstream/segmentation/configs/upernet_sw_tiny_512_160k_ade20k_ss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/segmentation/configs/upernet_sw_tiny_512_160k_ade20k_ss.py -------------------------------------------------------------------------------- /downstream/segmentation/configs/upernet_unireplknet_s_512_160k_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/segmentation/configs/upernet_unireplknet_s_512_160k_ade20k.py -------------------------------------------------------------------------------- /downstream/segmentation/mmseg/core/optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/segmentation/mmseg/core/optimizers/__init__.py -------------------------------------------------------------------------------- /downstream/segmentation/mmseg/core/optimizers/layer_decay_optimizer_constructor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/segmentation/mmseg/core/optimizers/layer_decay_optimizer_constructor.py -------------------------------------------------------------------------------- /downstream/segmentation/slak.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/segmentation/slak.py -------------------------------------------------------------------------------- /downstream/segmentation/unireplknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/downstream/segmentation/unireplknet.py -------------------------------------------------------------------------------- /engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/engine.py -------------------------------------------------------------------------------- /erf/SLaK_for_erf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/erf/SLaK_for_erf.py -------------------------------------------------------------------------------- /erf/SW_for_erf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/erf/SW_for_erf.py -------------------------------------------------------------------------------- /erf/Saved_matrix/ConvNext-T_31.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/erf/Saved_matrix/ConvNext-T_31.npy -------------------------------------------------------------------------------- /erf/Saved_matrix/ConvNext-T_7.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/erf/Saved_matrix/ConvNext-T_7.npy -------------------------------------------------------------------------------- /erf/Saved_matrix/SLaK-T_51.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/erf/Saved_matrix/SLaK-T_51.npy -------------------------------------------------------------------------------- /erf/Saved_matrix/SW.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/erf/Saved_matrix/SW.npy -------------------------------------------------------------------------------- /erf/analyze_erf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/erf/analyze_erf.py -------------------------------------------------------------------------------- /erf/erf_slak51_convnext7_convnext31.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/erf/erf_slak51_convnext7_convnext31.py -------------------------------------------------------------------------------- /erf/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/erf/readme.md -------------------------------------------------------------------------------- /erf/resnet_for_erf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/erf/resnet_for_erf.py -------------------------------------------------------------------------------- /erf/sw.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/erf/sw.pdf -------------------------------------------------------------------------------- /erf/sw_00.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/erf/sw_00.png -------------------------------------------------------------------------------- /erf/visualize_erf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/erf/visualize_erf.py -------------------------------------------------------------------------------- /flopses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/flopses.py -------------------------------------------------------------------------------- /funcs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/funcs.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/main.py -------------------------------------------------------------------------------- /model_sema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/model_sema.py -------------------------------------------------------------------------------- /models/SLaK.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/models/SLaK.py -------------------------------------------------------------------------------- /models/SLaK_pw_wt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/models/SLaK_pw_wt.py -------------------------------------------------------------------------------- /models/SLaK_reg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/models/SLaK_reg.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/taichi_kernel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/models/taichi_kernel.py -------------------------------------------------------------------------------- /optim_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/optim_factory.py -------------------------------------------------------------------------------- /run_with_submitit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/run_with_submitit.py -------------------------------------------------------------------------------- /shiftadd/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/LICENSE -------------------------------------------------------------------------------- /shiftadd/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/README.md -------------------------------------------------------------------------------- /shiftadd/SLAK_se.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/SLAK_se.py -------------------------------------------------------------------------------- /shiftadd/SLaK_gr_sw_sf_se.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/SLaK_gr_sw_sf_se.py -------------------------------------------------------------------------------- /shiftadd/SLaK_gr_sw_sf_se2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/SLaK_gr_sw_sf_se2.py -------------------------------------------------------------------------------- /shiftadd/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/compare.py -------------------------------------------------------------------------------- /shiftadd/compare_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/compare_module.py -------------------------------------------------------------------------------- /shiftadd/compare_mp_blur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/compare_mp_blur.py -------------------------------------------------------------------------------- /shiftadd/compare_mp_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/compare_mp_embedding.py -------------------------------------------------------------------------------- /shiftadd/compare_mp_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/compare_mp_linear.py -------------------------------------------------------------------------------- /shiftadd/compare_mp_sp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/compare_mp_sp.py -------------------------------------------------------------------------------- /shiftadd/is_jit_faster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/is_jit_faster.py -------------------------------------------------------------------------------- /shiftadd/ops/__init__.py: -------------------------------------------------------------------------------- 1 | from .ops_py import * -------------------------------------------------------------------------------- /shiftadd/ops/ops_py/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/ops/ops_py/__init__.py -------------------------------------------------------------------------------- /shiftadd/ops/ops_py/add_shift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/ops/ops_py/add_shift.py -------------------------------------------------------------------------------- /shiftadd/ops/src/addshift_kernel_v0.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/ops/src/addshift_kernel_v0.cu -------------------------------------------------------------------------------- /shiftadd/ops/src/addshift_kernel_vmp.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/ops/src/addshift_kernel_vmp.cu -------------------------------------------------------------------------------- /shiftadd/ops/src/addshift_kernel_vmp_blur.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/ops/src/addshift_kernel_vmp_blur.cu -------------------------------------------------------------------------------- /shiftadd/ops/src/addshift_kernel_vmp_linear.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/ops/src/addshift_kernel_vmp_linear.cu -------------------------------------------------------------------------------- /shiftadd/ops/src/addshift_kernel_vmp_pos_em.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/ops/src/addshift_kernel_vmp_pos_em.cu -------------------------------------------------------------------------------- /shiftadd/ops/src/addshift_ops_v0.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/ops/src/addshift_ops_v0.cpp -------------------------------------------------------------------------------- /shiftadd/ops/src/addshift_ops_vmp.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/ops/src/addshift_ops_vmp.cpp -------------------------------------------------------------------------------- /shiftadd/ops/src/addshift_ops_vmp_blur.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/ops/src/addshift_ops_vmp_blur.cpp -------------------------------------------------------------------------------- /shiftadd/ops/src/addshift_ops_vmp_linear.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/ops/src/addshift_ops_vmp_linear.cpp -------------------------------------------------------------------------------- /shiftadd/ops/src/addshift_ops_vmp_pos_em.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/ops/src/addshift_ops_vmp_pos_em.cpp -------------------------------------------------------------------------------- /shiftadd/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/setup.py -------------------------------------------------------------------------------- /shiftadd/test_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/shiftadd/test_ops.py -------------------------------------------------------------------------------- /sparse_core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/sparse_core.py -------------------------------------------------------------------------------- /test/depthwise_conv2d_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/test/depthwise_conv2d_test.py -------------------------------------------------------------------------------- /throughout/a100fps.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/throughout/a100fps.json -------------------------------------------------------------------------------- /throughout/models/SLaK_gemm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/throughout/models/SLaK_gemm.py -------------------------------------------------------------------------------- /throughout/models/SLaK_general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/throughout/models/SLaK_general.py -------------------------------------------------------------------------------- /throughout/models/SW_v2_unirep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/throughout/models/SW_v2_unirep.py -------------------------------------------------------------------------------- /throughout/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /throughout/models/dw_gemm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/throughout/models/dw_gemm.py -------------------------------------------------------------------------------- /throughout/models/unireplknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/throughout/models/unireplknet.py -------------------------------------------------------------------------------- /throughout/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/throughout/readme.md -------------------------------------------------------------------------------- /throughout/throughput.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/throughout/throughput.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lidc54/shift-wiseConv/HEAD/utils.py --------------------------------------------------------------------------------