├── LICENSE ├── README.md ├── ResNet-18-var ├── imagenet-resnet-transWino-prune.py ├── prune_script.py ├── prune_sh.sh ├── prune_test_sh.sh ├── retrain_sh.sh ├── view_model.py ├── view_sh.sh ├── winograd_conv.py └── winograd_imtrans.py ├── doc └── teaser.png └── winograd2x2_cublas ├── Makefile ├── winograd2x2_conv ├── Makefile ├── __init__.py ├── winograd2x2_conv.py ├── winograd2x2_conv_cuda.cu.cc ├── winograd2x2_conv_grad_cuda.cu.cc ├── winograd2x2_conv_grad_op.cc ├── winograd2x2_conv_grad_test.py ├── winograd2x2_conv_op.cc └── winograd2x2_conv_test.py └── winograd2x2_imTrans ├── Makefile ├── __init__.py ├── winograd2x2_imTrans.py ├── winograd2x2_imTrans_cuda.cu.cc ├── winograd2x2_imTrans_grad_cuda.cu.cc ├── winograd2x2_imTrans_grad_cuda.cu.cc.atomicVersion ├── winograd2x2_imTrans_grad_op.cc ├── winograd2x2_imTrans_grad_test.py ├── winograd2x2_imTrans_op.cc └── winograd2x2_imTrans_test.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/README.md -------------------------------------------------------------------------------- /ResNet-18-var/imagenet-resnet-transWino-prune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/ResNet-18-var/imagenet-resnet-transWino-prune.py -------------------------------------------------------------------------------- /ResNet-18-var/prune_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/ResNet-18-var/prune_script.py -------------------------------------------------------------------------------- /ResNet-18-var/prune_sh.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/ResNet-18-var/prune_sh.sh -------------------------------------------------------------------------------- /ResNet-18-var/prune_test_sh.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/ResNet-18-var/prune_test_sh.sh -------------------------------------------------------------------------------- /ResNet-18-var/retrain_sh.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/ResNet-18-var/retrain_sh.sh -------------------------------------------------------------------------------- /ResNet-18-var/view_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/ResNet-18-var/view_model.py -------------------------------------------------------------------------------- /ResNet-18-var/view_sh.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/ResNet-18-var/view_sh.sh -------------------------------------------------------------------------------- /ResNet-18-var/winograd_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/ResNet-18-var/winograd_conv.py -------------------------------------------------------------------------------- /ResNet-18-var/winograd_imtrans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/ResNet-18-var/winograd_imtrans.py -------------------------------------------------------------------------------- /doc/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/doc/teaser.png -------------------------------------------------------------------------------- /winograd2x2_cublas/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/winograd2x2_cublas/Makefile -------------------------------------------------------------------------------- /winograd2x2_cublas/winograd2x2_conv/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/winograd2x2_cublas/winograd2x2_conv/Makefile -------------------------------------------------------------------------------- /winograd2x2_cublas/winograd2x2_conv/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /winograd2x2_cublas/winograd2x2_conv/winograd2x2_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/winograd2x2_cublas/winograd2x2_conv/winograd2x2_conv.py -------------------------------------------------------------------------------- /winograd2x2_cublas/winograd2x2_conv/winograd2x2_conv_cuda.cu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/winograd2x2_cublas/winograd2x2_conv/winograd2x2_conv_cuda.cu.cc -------------------------------------------------------------------------------- /winograd2x2_cublas/winograd2x2_conv/winograd2x2_conv_grad_cuda.cu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/winograd2x2_cublas/winograd2x2_conv/winograd2x2_conv_grad_cuda.cu.cc -------------------------------------------------------------------------------- /winograd2x2_cublas/winograd2x2_conv/winograd2x2_conv_grad_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/winograd2x2_cublas/winograd2x2_conv/winograd2x2_conv_grad_op.cc -------------------------------------------------------------------------------- /winograd2x2_cublas/winograd2x2_conv/winograd2x2_conv_grad_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/winograd2x2_cublas/winograd2x2_conv/winograd2x2_conv_grad_test.py -------------------------------------------------------------------------------- /winograd2x2_cublas/winograd2x2_conv/winograd2x2_conv_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/winograd2x2_cublas/winograd2x2_conv/winograd2x2_conv_op.cc -------------------------------------------------------------------------------- /winograd2x2_cublas/winograd2x2_conv/winograd2x2_conv_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/winograd2x2_cublas/winograd2x2_conv/winograd2x2_conv_test.py -------------------------------------------------------------------------------- /winograd2x2_cublas/winograd2x2_imTrans/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/winograd2x2_cublas/winograd2x2_imTrans/Makefile -------------------------------------------------------------------------------- /winograd2x2_cublas/winograd2x2_imTrans/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /winograd2x2_cublas/winograd2x2_imTrans/winograd2x2_imTrans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/winograd2x2_cublas/winograd2x2_imTrans/winograd2x2_imTrans.py -------------------------------------------------------------------------------- /winograd2x2_cublas/winograd2x2_imTrans/winograd2x2_imTrans_cuda.cu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/winograd2x2_cublas/winograd2x2_imTrans/winograd2x2_imTrans_cuda.cu.cc -------------------------------------------------------------------------------- /winograd2x2_cublas/winograd2x2_imTrans/winograd2x2_imTrans_grad_cuda.cu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/winograd2x2_cublas/winograd2x2_imTrans/winograd2x2_imTrans_grad_cuda.cu.cc -------------------------------------------------------------------------------- /winograd2x2_cublas/winograd2x2_imTrans/winograd2x2_imTrans_grad_cuda.cu.cc.atomicVersion: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/winograd2x2_cublas/winograd2x2_imTrans/winograd2x2_imTrans_grad_cuda.cu.cc.atomicVersion -------------------------------------------------------------------------------- /winograd2x2_cublas/winograd2x2_imTrans/winograd2x2_imTrans_grad_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/winograd2x2_cublas/winograd2x2_imTrans/winograd2x2_imTrans_grad_op.cc -------------------------------------------------------------------------------- /winograd2x2_cublas/winograd2x2_imTrans/winograd2x2_imTrans_grad_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/winograd2x2_cublas/winograd2x2_imTrans/winograd2x2_imTrans_grad_test.py -------------------------------------------------------------------------------- /winograd2x2_cublas/winograd2x2_imTrans/winograd2x2_imTrans_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/winograd2x2_cublas/winograd2x2_imTrans/winograd2x2_imTrans_op.cc -------------------------------------------------------------------------------- /winograd2x2_cublas/winograd2x2_imTrans/winograd2x2_imTrans_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyul/sparse-winograd-cnn/HEAD/winograd2x2_cublas/winograd2x2_imTrans/winograd2x2_imTrans_test.py --------------------------------------------------------------------------------