У Caffe есть возможность выбрать cuDNN в качестве внутреннего движка для конвора. слой. WorkspaceData - это память графического процессора, необходимая для выполнения указанного алгоритма свертки для cuDNN conv. слой. Когда я просматриваю исходный код Caffe, в частности, cudnn_conv_layer.hpp, cudnn_conv_layer. cpp и cudnn_conv_layer.cu. Я не нашел, где размещена переменная workspaceData. Я также использую команду "grep" для поиска этой переменной, но я до сих пор не знаю, где эта переменная расположена.
Мой вопрос: где рабочие области выделяют для cuDNN conv. слой? Кажется, что эта переменная не имеет выделения памяти. Я знаю, что должно быть что-то, что я пропустил. Пожалуйста, пролите свет на этот вопрос.