diff options
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. |