diff options
Diffstat (limited to 'utils/TypePrinter.h')
| -rw-r--r-- | utils/TypePrinter.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h index 21f9c0849c..f55f72a4b8 100644 --- a/utils/TypePrinter.h +++ b/utils/TypePrinter.h @@ -628,6 +628,9 @@ inline ::std::ostream &operator<<(::std::ostream &os, const ActivationLayerInfo: case ActivationLayerInfo::ActivationFunction::HARD_SWISH: os << "HARD_SWISH"; break; + case ActivationLayerInfo::ActivationFunction::SWISH: + os << "SWISH"; + break; case ActivationLayerInfo::ActivationFunction::GELU: os << "GELU"; break; |