Searched refs:num_requested_algos (Results 1 – 1 of 1) sorted by relevance
2417 const int num_requested_algos = 5; in GetCudnnConvolutionForwardAlgo() local2419 cudnnConvolutionFwdAlgoPerf_t perf_results[num_requested_algos]; in GetCudnnConvolutionForwardAlgo()2423 output_nd.handle(), num_requested_algos, &num_returned_algos, in GetCudnnConvolutionForwardAlgo()2458 const int num_requested_algos = 5; in GetCudnnConvolutionBackwardDataAlgo() local2460 cudnnConvolutionBwdDataAlgoPerf_t perf_results[num_requested_algos]; in GetCudnnConvolutionBackwardDataAlgo()2464 input_nd.handle(), num_requested_algos, &num_returned_algos, in GetCudnnConvolutionBackwardDataAlgo()2501 const int num_requested_algos = 5; in GetCudnnConvolutionBackwardFilterAlgo() local2503 cudnnConvolutionBwdFilterAlgoPerf_t perf_results[num_requested_algos]; in GetCudnnConvolutionBackwardFilterAlgo()2507 filter.handle(), num_requested_algos, &num_returned_algos, perf_results)); in GetCudnnConvolutionBackwardFilterAlgo()