3 ms·
http://arxiv.org/pdf/1412.7580v2.pdf http://arxiv.org/pdf/1412.7580v2.pdf compares the convolutional implementation with the cuDNN layers. For the FC layers, i
by ajtulloch 12y ago
http://arxiv.org/pdf/1412.7580v2.pdf http://arxiv.org/pdf/1412.7580v2.pdf compares the convolutional implementation with the cuDNN layers. For the FC layers, it's just CuBLAS `sgemm`.