Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dgpu_utils.cc32 tensorflow::CudnnVersion cudnn_version; in GetCudnnVersion() local
37 cudnn_version.set_major(version.major_version()); in GetCudnnVersion()
38 cudnn_version.set_minor(version.minor_version()); in GetCudnnVersion()
39 cudnn_version.set_patch(version.patch()); in GetCudnnVersion()
42 return cudnn_version; in GetCudnnVersion()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_algorithm_picker.cc101 tensorflow::CudnnVersion cudnn_version; in GetCudnnVersion() local
106 cudnn_version.set_major(version.major_version()); in GetCudnnVersion()
107 cudnn_version.set_minor(version.minor_version()); in GetCudnnVersion()
108 cudnn_version.set_patch(version.patch()); in GetCudnnVersion()
111 return cudnn_version; in GetCudnnVersion()
/external/tensorflow/tensorflow/stream_executor/cuda/
DBUILD282 ":cudnn_version",
436 name = "cudnn_version",
437 srcs = ["cudnn_version.cc"],
438 hdrs = ["cudnn_version.h"],
448 ":cudnn_version",
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc274 tensorflow::CudnnVersion cudnn_version; in GetCudnnVersion() local
279 cudnn_version.set_major(version.major_version()); in GetCudnnVersion()
280 cudnn_version.set_minor(version.minor_version()); in GetCudnnVersion()
281 cudnn_version.set_patch(version.patch()); in GetCudnnVersion()
284 return cudnn_version; in GetCudnnVersion()
/external/tensorflow/tensorflow/core/protobuf/
Dautotuning.proto71 CudnnVersion cudnn_version = 3; field