Searched defs:cudnn_dnn (Results 1 – 2 of 2) sorted by relevance
373 gpu::CudnnSupport *cudnn_dnn = dynamic_cast<gpu::CudnnSupport *>(dnn); in ConvolveWithExecutionPlan() local431 gpu::CudnnSupport *cudnn_dnn = dynamic_cast<gpu::CudnnSupport *>(dnn); in FusedConvolveWithExecutionPlan() local468 gpu::CudnnSupport *cudnn_dnn = dynamic_cast<gpu::CudnnSupport *>(dnn); in ConvolveBackwardDataWithExecutionPlan() local560 gpu::CudnnSupport *cudnn_dnn = dynamic_cast<gpu::CudnnSupport *>(dnn); in ConvolveBackwardFilterWithExecutionPlan() local
393 gpu::CudnnSupport *cudnn_dnn = in GetFusedConvolveExecutionPlans() local