Searched defs:cudnn_version (Results 1 – 12 of 12) sorted by relevance
56 tensorflow::CudnnVersion cudnn_version; in TEST_F() local74 tensorflow::CudnnVersion cudnn_version; in TEST_F() local
53 tensorflow::ComputeCapability cc, tensorflow::CudnnVersion cudnn_version,
26 tensorflow.CudnnVersion cudnn_version = 3; field
220 tensorflow::CudnnVersion cudnn_version; in GetCudnnVersion() local
256 se::port::StatusOr<se::dnn::VersionInfo> cudnn_version = in IsConvBF16Supported() local
34 int64_t cudnn_version() { in cudnn_version() function
109 AutoMixedPrecisionListsCuda(int cuda_version, int cudnn_version) in AutoMixedPrecisionListsCuda()
97 CudnnVersion cudnn_version = 3; field
112 tensorflow::CudnnVersion cudnn_version; in GetCudnnVersion() local
327 long cudnn_version = at::detail::getCUDAHooks().versionCuDNN(); in cudnn_conv_suggest_memory_format() local
428 static long cudnn_version = detail::getCUDAHooks().versionCuDNN(); in use_cudnn() local467 long cudnn_version = detail::getCUDAHooks().versionCuDNN(); in use_cudnn_depthwise() local
317 long cudnn_version = at::detail::getCUDAHooks().versionCuDNN(); in check_cudnn_tensor_shapes() local