diff options
Diffstat (limited to 'arm_compute/graph/TypePrinter.h')
| -rw-r--r-- | arm_compute/graph/TypePrinter.h | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/arm_compute/graph/TypePrinter.h b/arm_compute/graph/TypePrinter.h index 0ecd57de9d..6babd3961d 100644 --- a/arm_compute/graph/TypePrinter.h +++ b/arm_compute/graph/TypePrinter.h @@ -295,6 +295,24 @@ inline ::std::ostream &operator<<(::std::ostream &os, const ConvolutionMethod &m return os; } +/** Formatted output of the FastMathHint type. */ +inline ::std::ostream &operator<<(::std::ostream &os, const FastMathHint &hint) +{ + switch(hint) + { + case FastMathHint::ENABLED: + os << "ENABLED"; + break; + case FastMathHint::DISABLED: + os << "DISABLED"; + break; + default: + ARM_COMPUTE_ERROR("NOT_SUPPORTED!"); + } + + return os; +} + /** Formatted output of the DepthwiseConvolutionMethod type. */ inline ::std::ostream &operator<<(::std::ostream &os, const DepthwiseConvolutionMethod &method) { |