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.cc2875 class CudnnEnvVar { class
2955 if (CudnnEnvVar<RnnDoFP32ComputationFP16Input>::IsEnabled()) { in GetRnnComputeType()
2971 return CudnnEnvVar<ConvDoFP32ComputationFP16Input>::IsEnabled() in GetConvAccumulatorType()
3320 if (CudnnEnvVar<FftTilingForward>::IsEnabled()) { in GetConvolveAlgorithms()
3323 if (CudnnEnvVar<WinogradNonfused>::IsEnabled() && with_winograd_nonfused) { in GetConvolveAlgorithms()
3385 if (CudnnEnvVar<WinogradNonfused>::IsEnabled() && with_winograd_nonfused) { in GetConvolveBackwardDataAlgorithms()
3428 if (CudnnEnvVar<WinogradNonfused>::IsEnabled() && with_winograd_nonfused) { in GetConvolveBackwardFilterAlgorithms()