From bdff7e95a82c0f2b93ea9edda2996334af358294 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" <gardener@tensorflow.org> Date: Sun, 27 Dec 2020 19:40:58 -0800 Subject: [PATCH] add missing `ActivationMode::kNone` to `ActivationModeString` PiperOrigin-RevId: 349228265 Change-Id: Ia2429e4df6d7fe91ce712caa6c15eb9965b2b445 --- tensorflow/stream_executor/dnn.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tensorflow/stream_executor/dnn.cc b/tensorflow/stream_executor/dnn.cc index 6aba656fc68..bbc7f6a537f 100644 --- a/tensorflow/stream_executor/dnn.cc +++ b/tensorflow/stream_executor/dnn.cc @@ -94,6 +94,8 @@ std::string QuantizedActivationModeString(QuantizedActivationMode mode) { std::string ActivationModeString(ActivationMode mode) { switch (mode) { + case ActivationMode::kNone: + return "none"; case ActivationMode::kSigmoid: return "sigmoid"; case ActivationMode::kRelu: