From 391cdd80952e9cc546d82a8bf2fe7dd04f46cb2f Mon Sep 17 00:00:00 2001 From: Tim Shen Date: Mon, 24 Sep 2018 17:44:58 -0700 Subject: Add cuDNN fused convolution forward support. The tests are in the next patch. PiperOrigin-RevId: 214362688 --- tensorflow/stream_executor/dnn.h | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'tensorflow/stream_executor') 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. -- cgit v1.2.3