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 ed578307b5..0ecd57de9d 100644 --- a/arm_compute/graph/TypePrinter.h +++ b/arm_compute/graph/TypePrinter.h @@ -138,6 +138,24 @@ inline ::std::ostream &operator<<(::std::ostream &os, const Target &target) return os; } +/** Formatted output of the DataLayout */ +inline ::std::ostream &operator<<(::std::ostream &os, const DataLayout &data_layout) +{ + switch(data_layout) + { + case DataLayout::NCHW: + os << "NCHW"; + break; + case DataLayout::NHWC: + os << "NHWC"; + break; + default: + ARM_COMPUTE_ERROR("NOT_SUPPORTED!"); + } + + return os; +} + /** Formatted output of the activation function type. */ inline ::std::ostream &operator<<(::std::ostream &os, const ActivationLayerInfo::ActivationFunction &act_function) { |