search

josephjaspers / blackcat_tensors

Matrix-Vector Library Designed for Neural Network Construction. cuda (gpu) support, openmp (multithreaded cpu) support, partial support of BLAS, expression template based implementation PTX code generation identical to hand written kernels, and support for auto-differentiation
12 stars 4 forks source link

Neural_Networks: Add GPU support for Convolution #50

Closed josephjaspers closed 4 years ago

josephjaspers commented 4 years ago

Currently have a working forward_propagation for conv

josephjaspers commented 4 years ago

Forgot to update: This has been complete (gpu and cpu support) prior to: https://github.com/josephjaspers/blackcat_tensors/commit/95710e88162953c992310386b3e9cfe8969bf7d9