Какой файл выполняет caffe при выполнении сверточной операции? - PullRequest
Новая
       75

Какой файл выполняет caffe при выполнении сверточной операции?

0 голосов
/ 13 октября 2018

В caffe\src\layers есть три файла, связанных со сверткой: conv_layer.cpp, conv_layer.cu, cudnn_conv_layer.cpp. Какой файл оправдывает себя при выполнении сверточной операции?

1 Ответ

0 голосов
/ 05 декабря 2018

Для реализации ЦП: 

./src/caffe/layers/conv_layer.cpp

Для реализации CUDA GPU:

./src/caffe/layers/conv_layer.cu

...