diff options
Diffstat (limited to 'tensorflow/stream_executor/dnn.h')
| -rw-r--r-- | tensorflow/stream_executor/dnn.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/dnn.h b/tensorflow/stream_executor/dnn.h index 43cfd313c1..3c47d2c2e8 100644 --- a/tensorflow/stream_executor/dnn.h +++ b/tensorflow/stream_executor/dnn.h @@ -25,6 +25,7 @@ limitations under the License. #include <functional> #include <limits> #include <memory> +#include <tuple> #include "tensorflow/stream_executor/device_memory.h" #include "tensorflow/stream_executor/lib/array_slice.h" @@ -885,6 +886,12 @@ class DnnSupport { virtual port::Status Init() = 0; + // Gets the version of the backing library, as a {major, minor, patch} tuple. + virtual port::StatusOr<std::tuple<int, int, int>> GetVersion() { + return port::UnimplementedError( + "DnnSupport::GetVersion not implemented on this platform."); + } + // Performs a single-precision forward batch normalization operation onto // the stream. // |