Searched refs:cudnn_launcher (Results 1 – 1 of 1) sorted by relevance
6247 auto cudnn_launcher = [&](CudnnTensorDescriptor& src_desc, in DoPoolForward() local6279 cudnn_launcher(src_desc, dest_desc, input_data_ptr, output_data_ptr); in DoPoolForward()6314 auto cudnn_launcher = [&](CudnnTensorDescriptor& src_desc, in DoPoolBackward() local6351 const auto status = cudnn_launcher(src_desc, dest_desc, output_data_ptr, in DoPoolBackward()