Searched refs:CudnnEnvVar (Results 1 – 1 of 1) sorted by relevance
3243 class CudnnEnvVar { class3358 if (CudnnEnvVar<RnnDoFP32ComputationFP16Input>::IsEnabled()) { in GetRnnComputeType()3376 return CudnnEnvVar<ConvDoFP32ComputationFP16Input>::IsEnabled() in GetConvActivationType()3385 return CudnnEnvVar<ConvDoFP32ComputationFP16Input>::IsEnabled() in GetConvActivationType()3400 return CudnnEnvVar<ConvDoFP32ComputationFP16Input>::IsEnabled() in GetConvAccumulatorType()3408 return CudnnEnvVar<ConvDoFP32ComputationFP16Input>::IsEnabled() in GetConvAccumulatorType()4601 /*disable_winograd*/ !CudnnEnvVar<WinogradNonfused>::IsEnabled(), in CreateOpRunners()5294 if (CudnnEnvVar<FftTilingForward>::IsEnabled()) { in GetConvolveAlgorithms()5297 if (CudnnEnvVar<WinogradNonfused>::IsEnabled()) { in GetConvolveAlgorithms()5350 if (CudnnEnvVar<WinogradNonfused>::IsEnabled()) { in GetConvolveBackwardDataAlgorithms()[all …]