Searched refs:backward_algo (Results 1 – 1 of 1) sorted by relevance
287 int backward_algo = CUDNN_CONVOLUTION_BWD_DATA_PREFER_FASTEST; in cudnn_convolutional_setup() local292 backward_algo = CUDNN_CONVOLUTION_BWD_DATA_NO_WORKSPACE; in cudnn_convolutional_setup()299 backward_algo = CUDNN_CONVOLUTION_BWD_DATA_SPECIFY_WORKSPACE_LIMIT; in cudnn_convolutional_setup()317 (cudnnConvolutionBwdDataPreference_t)backward_algo, in cudnn_convolutional_setup()