search

Maratyszcza / NNPACK

Acceleration package for neural networks on multi-core CPUs
BSD 2-Clause "Simplified" License
1.67k stars 316 forks source link

3D Conv #138

Open al-rigazzi opened 6 years ago

al-rigazzi commented 6 years ago

I know this is beyond the initial purpose of this library, but I think many SDKs are missing a good optimized 3D convolution primitive. We see the need for this in many scientific applications, and it would be great if NNPACK got ahead of the competing libs.

ghost commented 6 years ago

Have a look at Intel mkl-dnn, they support 3D convolution and pooling.