caffe做卷积运算时执行哪个文件?
Which file does caffe execute when doing convolutional operation?
caffe\src\layers
中有3个与卷积相关的文件:conv_layer.cpp
、conv_layer.cu
、cudnn_conv_layer.cpp
。caffe在执行卷积运算时执行的是哪个文件?
对于 CPU 实施:
./src/caffe/layers/conv_layer.cpp
对于 CUDA GPU 实现:
./src/caffe/layers/conv_layer.cu
caffe\src\layers
中有3个与卷积相关的文件:conv_layer.cpp
、conv_layer.cu
、cudnn_conv_layer.cpp
。caffe在执行卷积运算时执行的是哪个文件?
对于 CPU 实施:
./src/caffe/layers/conv_layer.cpp
对于 CUDA GPU 实现:
./src/caffe/layers/conv_layer.cu