diff options
Diffstat (limited to 'utils/GraphTypePrinter.h')
-rw-r--r-- | utils/GraphTypePrinter.h | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/utils/GraphTypePrinter.h b/utils/GraphTypePrinter.h index be56d59853..4ff1019cce 100644 --- a/utils/GraphTypePrinter.h +++ b/utils/GraphTypePrinter.h @@ -58,6 +58,31 @@ inline std::string to_string(const ConvolutionMethodHint &conv_method) str << conv_method; return str.str(); } + +/** Formatted output of the @ref TargetHint type. */ +inline ::std::ostream &operator<<(::std::ostream &os, const TargetHint &target_hint) +{ + switch(target_hint) + { + case TargetHint::NEON: + os << "NEON"; + break; + case TargetHint::OPENCL: + os << "OPENCL"; + break; + default: + ARM_COMPUTE_ERROR("NOT_SUPPORTED!"); + } + + return os; +} + +inline std::string to_string(const TargetHint &target_hint) +{ + std::stringstream str; + str << target_hint; + return str.str(); +} } // namespace graph } // namespace arm_compute #endif /* __ARM_COMPUTE_TEST_GRAPH_TYPE_PRINTER_H__ */ |