projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: e7ea87f) | patch
[XLA:GPU] Eliminate the guard around Winograd non-fused convolutions with cudnn7.
authorJustin Lebar <jlebar@google.com>
Sat, 7 Apr 2018 18:42:43 +0000 (11:42 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Sat, 7 Apr 2018 18:45:04 +0000 (11:45 -0700)
commit1cd76c209ce6f74298843568a7fc397c2e6f958f
treec4647ef54eaba837b9a5a1a05b0cf029aaec7b36tree | snapshot
parente7ea87f97e03360719d132a71acc1eb2f93c249fcommit | diff
[XLA:GPU] Eliminate the guard around Winograd non-fused convolutions with cudnn7.

Adds DnnSupport::GetVersion() and uses this to unguard Winograd
non-fused convolutions if you're using cudnn7.

PiperOrigin-RevId: 192010450
tensorflow/compiler/xla/service/gpu/cudnn_convolution_algorithm_picker.cc diff | blob | history
tensorflow/stream_executor/cuda/cuda_dnn.cc diff | blob | history
tensorflow/stream_executor/cuda/cuda_dnn.h diff | blob | history
tensorflow/stream_executor/dnn.h diff | blob | history
Domain: Machine Learning / ML Framework;