Home
last modified time | relevance | path

Searched refs:CudnnEnvVar (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc2491 class CudnnEnvVar { class
2569 if (CudnnEnvVar<RnnDoFP32ComputationFP16Input>::IsEnabled()) { in GetRnnComputeType()
2585 return CudnnEnvVar<ConvDoFP32ComputationFP16Input>::IsEnabled() in GetConvAccumulatorType()
3064 if (CudnnEnvVar<FftTilingForward>::IsEnabled()) { in GetConvolveAlgorithms()
3067 if (CudnnEnvVar<WinogradNonfused>::IsEnabled() && with_winograd_nonfused) { in GetConvolveAlgorithms()
3124 if (CudnnEnvVar<WinogradNonfused>::IsEnabled() && with_winograd_nonfused) { in GetConvolveBackwardDataAlgorithms()
3164 if (CudnnEnvVar<WinogradNonfused>::IsEnabled() && with_winograd_nonfused) { in GetConvolveBackwardFilterAlgorithms()