diff options
Diffstat (limited to 'tensorflow/stream_executor/cuda/cuda_dnn.h')
| -rw-r--r-- | tensorflow/stream_executor/cuda/cuda_dnn.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/cuda/cuda_dnn.h b/tensorflow/stream_executor/cuda/cuda_dnn.h index e40ba9b012..0e5368aca8 100644 --- a/tensorflow/stream_executor/cuda/cuda_dnn.h +++ b/tensorflow/stream_executor/cuda/cuda_dnn.h @@ -46,6 +46,7 @@ class CudnnSupport : public dnn::DnnSupport { ~CudnnSupport() override; port::Status Init() override; + port::StatusOr<std::tuple<int, int, int>> GetVersion() override; port::StatusOr<std::unique_ptr<dnn::RnnDescriptor>> createRnnDescriptor( int num_layers, int hidden_size, int input_size, |