Searched refs:cudnn_launch_status (Results 1 – 8 of 8) sorted by relevance
420 auto cudnn_launch_status = stream->ConvolveBackwardDataWithAlgorithm( in operator ()() local424 if (cudnn_launch_status.ok() && profile_result.is_valid()) { in operator ()()500 auto cudnn_launch_status = stream->ConvolveBackwardDataWithAlgorithm( in operator ()() local505 if (!cudnn_launch_status.ok()) { in operator ()()506 ctx->SetStatus(cudnn_launch_status); in operator ()()
540 auto cudnn_launch_status = stream->ConvolveWithAlgorithm( in launch() local544 if (cudnn_launch_status.ok()) { in launch()625 auto cudnn_launch_status = stream->ConvolveWithAlgorithm( in launch() local629 if (!cudnn_launch_status.ok()) { in launch()630 ctx->SetStatus(cudnn_launch_status); in launch()
1454 auto cudnn_launch_status = stream->ConvolveBackwardDataWithAlgorithm( in Compute() local1458 if (cudnn_launch_status.ok()) { in Compute()1523 auto cudnn_launch_status = stream->ConvolveBackwardDataWithAlgorithm( in Compute() local1528 if (!cudnn_launch_status.ok()) { in Compute()1529 context->SetStatus(cudnn_launch_status); in Compute()1938 auto cudnn_launch_status = stream->ConvolveBackwardFilterWithAlgorithm( in Compute() local1942 if (cudnn_launch_status.ok()) { in Compute()1969 auto cudnn_launch_status = stream->ConvolveBackwardFilterWithAlgorithm( in Compute()1974 if (cudnn_launch_status.ok()) { in Compute()2016 auto cudnn_launch_status = stream->ConvolveBackwardFilterWithAlgorithm( in Compute() local[all …]
350 bool cudnn_launch_status = in Compute() local357 if (!cudnn_launch_status) { in Compute()
405 Status cudnn_launch_status =409 if (cudnn_launch_status.ok() && profile_result.is_valid()) {701 Status cudnn_launch_status = launch(algorithm_config, &scratch_allocator,703 OP_REQUIRES_OK(context, cudnn_launch_status);
1015 auto cudnn_launch_status = stream->ConvolveBackwardFilterWithAlgorithm( in operator ()() local1019 if (cudnn_launch_status.ok() && profile_result.is_valid()) { in operator ()()1100 auto cudnn_launch_status = stream->ConvolveBackwardFilterWithAlgorithm( in operator ()() local1105 if (!cudnn_launch_status.ok()) { in operator ()()1106 ctx->SetStatus(cudnn_launch_status); in operator ()()
1027 auto cudnn_launch_status = stream->ConvolveWithAlgorithm( in operator ()() local1031 if (cudnn_launch_status.ok() && profile_result.is_valid()) { in operator ()()1116 auto cudnn_launch_status = stream->ConvolveWithAlgorithm( in operator ()() local1120 if (!cudnn_launch_status.ok()) { in operator ()()1121 ctx->SetStatus(cudnn_launch_status); in operator ()()
954 bool cudnn_launch_status = in operator ()() local967 if (!cudnn_launch_status) { in operator ()()1126 bool cudnn_launch_status = in operator ()() local1135 if (!cudnn_launch_status) { in operator ()()