diff options
| author |  Tim Shen <timshen@google.com> | 2018-09-24 17:44:58 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-24 17:48:32 -0700 |
| commit | 391cdd80952e9cc546d82a8bf2fe7dd04f46cb2f (patch) | |
| tree | a253b5bd01d2088d07e1cae505028a600d834384 /tensorflow/stream_executor | |
| parent | 9ab01c6732dae1143e22713375a9cc7758216787 (diff) | |
Add cuDNN fused convolution forward support.
The tests are in the next patch.
PiperOrigin-RevId: 214362688
Diffstat (limited to 'tensorflow/stream_executor')
| -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 9abfa1db6a..621b155240 100644 --- a/tensorflow/stream_executor/dnn.h +++ b/tensorflow/stream_executor/dnn.h @@ -873,7 +873,7 @@ class NormalizeDescriptor { // Describes a kind of non-linearity (threshold-like mathematical function). enum class ActivationMode { - kNone, + kNone = 0, kSigmoid, // Rectified linear activation: f(x) = x < 0 ? 0 : x kRelu, @@ -885,6 +885,8 @@ enum class ActivationMode { kTanh, // Like ReluX, but passes all values in the range [-X,X]. kBandPass, + + kNumActivationModes, // Always in the end. }; // Returns a string representation of the given activation mode. |