diff options
author | Xiaoqiang Zheng <zhengxq@google.com> | 2016-05-07 22:22:44 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-05-09 09:21:14 -0700 |
commit | 6ff265ebae14586db5db623b7502ddbdbc8bbd12 (patch) | |
tree | d4df06ea887dfb603b22c9df1147b41b38994b54 /tensorflow/stream_executor/stream_executor_pimpl.h | |
parent | a846576a67da4bdd0b610a17b5c8d0d92e41f094 (diff) |
Adding autotune to the Cudnn conv algorithm selection.
For now, use TF_CUDNN_USE_AUTOTUNE=1 to enable this feature. Once it is mature
enough, it will be turned on by default.
Support for the backward steps will be added later.
Change: 121769364
Diffstat (limited to 'tensorflow/stream_executor/stream_executor_pimpl.h')
-rw-r--r-- | tensorflow/stream_executor/stream_executor_pimpl.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/stream_executor_pimpl.h b/tensorflow/stream_executor/stream_executor_pimpl.h index f624e0fcdb..31b110a8e0 100644 --- a/tensorflow/stream_executor/stream_executor_pimpl.h +++ b/tensorflow/stream_executor/stream_executor_pimpl.h @@ -338,6 +338,9 @@ class StreamExecutor { // platform that underlies this interface. bool SupportsDnn() const; + // Get the list of supported algorithms for the forward convolution opeartion. + bool GetConvolveAlgorithms(std::vector<dnn::AlgorithmType> *out_algorithms); + // Returns the device ordinal that this StreamExecutor was initialized with. // Meaningless before initialization. int device_ordinal() const { return device_ordinal_; } |