experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add missing ActivationMode::kNone to ActivationModeString

Browse Source 
PiperOrigin-RevId: 349228265
Change-Id: Ia2429e4df6d7fe91ce712caa6c15eb9965b2b445
This commit is contained in:
A. Unique TensorFlower 2020-12-27 19:40:58 -08:00 committed by TensorFlower Gardener
parent f9fd71a45e
commit bdff7e95a8
1 changed files with 2 additions and 0 deletions

2
tensorflow/stream_executor/dnn.cc
View File

@ -94,6 +94,8 @@ std::string QuantizedActivationModeString(QuantizedActivationMode mode) {
std::string ActivationModeString(ActivationMode mode) { std::string ActivationModeString(ActivationMode mode) {
switch (mode) { switch (mode) {
case ActivationMode::kNone:
return "none";
case ActivationMode::kSigmoid: case ActivationMode::kSigmoid:
return "sigmoid"; return "sigmoid";
case ActivationMode::kRelu: case ActivationMode::kRelu: