Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcudnn_version.cc22 CudnnVersion loaded_version) { in IsSourceCompatibleWithCudnnLibrary() argument
31 if (loaded_version.major_version != source_version.major_version) { in IsSourceCompatibleWithCudnnLibrary()
34 return ((loaded_version.minor_version == source_version.minor_version) || in IsSourceCompatibleWithCudnnLibrary()
36 loaded_version.minor_version >= source_version.minor_version)); in IsSourceCompatibleWithCudnnLibrary()
Dcudnn_version.h45 CudnnVersion loaded_version);
Dcuda_dnn.cc307 CudnnVersion loaded_version; in Init() local
308 TF_RETURN_IF_ERROR(GetLoadedCudnnVersion(&loaded_version)); in Init()
309 if (!IsSourceCompatibleWithCudnnLibrary(source_version, loaded_version)) { in Init()
311 "Loaded runtime CuDNN library: ", loaded_version.ToString(), in Init()
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py646 loaded_version = get_loaded_tensorrt_version()
648 tf_logging.info("Loaded TensorRT version: %s" % str(loaded_version))
650 if loaded_version[0] < compiled_version[0]:
655 ".".join([str(x) for x in loaded_version])) +
659 for i in zip(loaded_version, compiled_version):
664 ".".join([str(x) for x in loaded_version])))
669 ".".join([str(x) for x in loaded_version]))