diff options
Diffstat (limited to 'nn/common/TypeUtils.cpp')
-rw-r--r-- | nn/common/TypeUtils.cpp | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/nn/common/TypeUtils.cpp b/nn/common/TypeUtils.cpp index ffe2912c3..15b377b87 100644 --- a/nn/common/TypeUtils.cpp +++ b/nn/common/TypeUtils.cpp @@ -584,6 +584,20 @@ std::ostream& operator<<(std::ostream& os, const ErrorStatus& errorStatus) { return os << "ErrorStatus{" << underlyingType(errorStatus) << "}"; } +std::ostream& operator<<(std::ostream& os, const FusedActivationFunc& activation) { + switch (activation) { + case FusedActivationFunc::NONE: + return os << "NONE"; + case FusedActivationFunc::RELU: + return os << "RELU"; + case FusedActivationFunc::RELU1: + return os << "RELU1"; + case FusedActivationFunc::RELU6: + return os << "RELU6"; + } + return os << "FusedActivationFunc{" << underlyingType(activation) << "}"; +} + std::ostream& operator<<(std::ostream& os, const OutputShape& outputShape) { return os << "OutputShape{.dimensions=" << outputShape.dimensions << ", .isSufficient=" << (outputShape.isSufficient ? "true" : "false") << "}"; |