The repository contains convolution implementation in C++. The python script is used for validation
- g++
- python3
- torch
- numpy
To run convolution_nchw.cpp
./conv.sh
To run convolution_nhwc.cpp
./conv_nhwc.sh
To run convolution_shape_nhwc.cpp
./conv_shape_nhwc.sh