diff options
Diffstat (limited to 'tests/TypePrinter.h')
| -rw-r--r-- | tests/TypePrinter.h | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/tests/TypePrinter.h b/tests/TypePrinter.h index ff9863e1fb..c4f3495761 100644 --- a/tests/TypePrinter.h +++ b/tests/TypePrinter.h @@ -197,9 +197,6 @@ inline ::std::ostream &operator<<(::std::ostream &os, const ActivationLayerInfo: case ActivationLayerInfo::ActivationFunction::ABS: os << "ABS"; break; - case ActivationLayerInfo::ActivationFunction::BOUNDED_RELU: - os << "BOUNDED_RELU"; - break; case ActivationLayerInfo::ActivationFunction::LINEAR: os << "LINEAR"; break; @@ -209,6 +206,12 @@ inline ::std::ostream &operator<<(::std::ostream &os, const ActivationLayerInfo: case ActivationLayerInfo::ActivationFunction::RELU: os << "RELU"; break; + case ActivationLayerInfo::ActivationFunction::BOUNDED_RELU: + os << "BOUNDED_RELU"; + break; + case ActivationLayerInfo::ActivationFunction::LEAKY_RELU: + os << "LEAKY_RELU"; + break; case ActivationLayerInfo::ActivationFunction::SOFT_RELU: os << "SOFT_RELU"; break; |