diff options
author | 2017-01-24 10:05:12 -0800 | |
---|---|---|
committer | 2017-01-24 10:27:38 -0800 | |
commit | eb65eff6cc65875c67e7dbf1c2e63baa6664bb6f (patch) | |
tree | 7e92646fb31da8823f622e129e1199a31c89aead /tensorflow/stream_executor/dnn.h | |
parent | 7f79424f63c5684a43d47de216ae152144ddeecf (diff) |
Add options argument for DNN activation
This is useful for platform-dependent functionality.
Change: 145432435
Diffstat (limited to 'tensorflow/stream_executor/dnn.h')
-rw-r--r-- | tensorflow/stream_executor/dnn.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/stream_executor/dnn.h b/tensorflow/stream_executor/dnn.h index 517f3ea904..6a3e8d0685 100644 --- a/tensorflow/stream_executor/dnn.h +++ b/tensorflow/stream_executor/dnn.h @@ -1357,7 +1357,9 @@ class DnnSupport { virtual bool DoActivate(Stream* stream, ActivationMode activation_mode, const BatchDescriptor& dimensions, const DeviceMemory<float>& input_data, - DeviceMemory<float>* output_data) = 0; + DeviceMemory<float>* output_data, uint64 options) { + return false; + } // Concatenates several layers into one, by concatenating the depth of each // layer at matching x and y coordinates. |