Add options argument for DNN activation

This is useful for platform-dependent functionality.
Change: 145432435

1 files changed, 11 insertions, 2 deletions

diff --git a/tensorflow/stream_executor/stream.cc b/tensorflow/stream_executor/stream.cc
index 7712a3697c..44c6d70636 100644
--- a/tensorflow/stream_executor/stream.cc
+++ b/tensorflow/stream_executor/stream.cc
@@ -1106,13 +1106,22 @@ Stream &Stream::ThenActivate(dnn::ActivationMode activation_mode,
                              const dnn::BatchDescriptor &dimensions,
                              const DeviceMemory<float> &input_data,
                              DeviceMemory<float> *output_data) {
+  return ThenActivateWithOptions(activation_mode, dimensions, input_data,
+                                 output_data, /*options=*/0);
+}
+
+Stream &Stream::ThenActivateWithOptions(dnn::ActivationMode activation_mode,
+                                        const dnn::BatchDescriptor &dimensions,
+                                        const DeviceMemory<float> &input_data,
+                                        DeviceMemory<float> *output_data,
+                                        uint64 options) {
   VLOG_CALL(PARAM(activation_mode), PARAM(dimensions), PARAM(input_data),
-            PARAM(output_data));
+            PARAM(output_data), PARAM(options));
 
   if (ok()) {
     if (dnn::DnnSupport *dnn = parent_->AsDnn()) {
       CheckError(dnn->DoActivate(this, activation_mode, dimensions, input_data,
-                                 output_data));
+                                 output_data, options));
     } else {
       SetErrorAndLogNoDnnSupport();
     }