summaryrefslogtreecommitdiff
path: root/nn/common/TypeUtils.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'nn/common/TypeUtils.cpp')
-rw-r--r--nn/common/TypeUtils.cpp14
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") << "}";