Searched refs:fwdAlgPerf (Results 1 – 1 of 1) sorted by relevance
743 .try_all([&](const cudnnConvolutionFwdAlgoPerf_t& fwdAlgPerf) { in raw_cudnn_convolution_forward_out_32bit() argument744 Tensor workspace = allocate_workspace(fwdAlgPerf.memory, input); in raw_cudnn_convolution_forward_out_32bit()749 ASSERT_CORRECT_PRECISION(fwdAlgPerf.mathType); in raw_cudnn_convolution_forward_out_32bit()752 args.cdesc.mut_desc(), fwdAlgPerf.mathType), in raw_cudnn_convolution_forward_out_32bit()767 fwdAlgPerf.algo, in raw_cudnn_convolution_forward_out_32bit()769 fwdAlgPerf.memory, in raw_cudnn_convolution_forward_out_32bit()775 static_cast<int>(fwdAlgPerf.algo), in raw_cudnn_convolution_forward_out_32bit()1156 .try_all([&](const cudnnConvolutionFwdAlgoPerf_t& fwdAlgPerf) { in raw_cudnn_convolution_add_relu_out_v7() argument1157 Tensor workspace = allocate_workspace(fwdAlgPerf.memory, input); in raw_cudnn_convolution_add_relu_out_v7()1162 ASSERT_CORRECT_PRECISION(fwdAlgPerf.mathType); in raw_cudnn_convolution_add_relu_out_v7()[all …]