Searched refs:GetCudnnVersion (Results 1 – 4 of 4) sorted by relevance
35 string GetCudnnVersion() { return TF_CUDNN_VERSION; } in GetCudnnVersion() function117 return GetDsoHandle("cudnn", GetCudnnVersion()); in GetCudnnDsoHandle()
99 tensorflow::CudnnVersion GetCudnnVersion(se::StreamExecutor* stream_executor) { in GetCudnnVersion() function152 *log.mutable_cudnn_version() = GetCudnnVersion(stream_exec); in LogConvAutotuneResults()199 *log.mutable_cudnn_version() = GetCudnnVersion(stream_exec); in LogFusedConvForwardAutotuneResults()
164 tensorflow::CudnnVersion GetCudnnVersion(se::StreamExecutor* stream_executor) { in GetCudnnVersion() function446 GetComputeCapability(stream_exec_), GetCudnnVersion(stream_exec_), in PickBestAlgorithmNoCacheCuda()515 *entry->mutable_cudnn_version() = GetCudnnVersion(stream_exec_); in PickBestAlgorithmNoCacheCuda()594 *log.mutable_cudnn_version() = GetCudnnVersion(stream_exec_); in PickBestAlgorithmNoCacheCuda()
910 int GetCudnnVersion(const Cluster& cluster) { in GetCudnnVersion() function939 cudnn_version_(GetCudnnVersion(*cluster)), in AutoMixedPrecisionImpl()