Searched refs:num_returned_algos (Results 1 – 1 of 1) sorted by relevance
2418 int num_returned_algos = 0; in GetCudnnConvolutionForwardAlgo() local2423 output_nd.handle(), num_requested_algos, &num_returned_algos, in GetCudnnConvolutionForwardAlgo()2427 for (int r = 0; r < num_returned_algos; r++) { in GetCudnnConvolutionForwardAlgo()2459 int num_returned_algos = 0; in GetCudnnConvolutionBackwardDataAlgo() local2464 input_nd.handle(), num_requested_algos, &num_returned_algos, in GetCudnnConvolutionBackwardDataAlgo()2468 for (int r = 0; r < num_returned_algos; r++) { in GetCudnnConvolutionBackwardDataAlgo()2502 int num_returned_algos = 0; in GetCudnnConvolutionBackwardFilterAlgo() local2507 filter.handle(), num_requested_algos, &num_returned_algos, perf_results)); in GetCudnnConvolutionBackwardFilterAlgo()2510 for (int r = 0; r < num_returned_algos; r++) { in GetCudnnConvolutionBackwardFilterAlgo()