diff options
Diffstat (limited to 'utils')
-rw-r--r-- | utils/TypePrinter.h | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h index c3af0a2419..9b9c7b5b34 100644 --- a/utils/TypePrinter.h +++ b/utils/TypePrinter.h @@ -3678,6 +3678,35 @@ inline std::string to_string(const experimental::dynamic_fusion::SoftmaxAttribut str << softmax_attr; return str.str(); } +/** Formatted output of the arm_compute::MatMulInfo type. + * + * @param[out] os Output stream. + * @param[in] matmul_info arm_compute::MatMulInfo type to output. + * + * @return Modified output stream. + */ +inline ::std::ostream &operator<<(::std::ostream &os, const arm_compute::MatMulInfo &matmul_info) +{ + os << "MatMulKernelInfo=" + << "[" + << "adj_lhs=" << matmul_info.adj_lhs() << ", " + << "adj_rhs=" << matmul_info.adj_rhs() << ", " + << "fused_activation=" << matmul_info.fused_activation().activation() << "]"; + + return os; +} +/** Formatted output of the arm_compute::MatMulInfo type. + * + * @param[in] matmul_info arm_compute::MatMulInfo type to output. + * + * @return Formatted string. + */ +inline std::string to_string(const arm_compute::MatMulInfo &matmul_info) +{ + std::stringstream str; + str << matmul_info; + return str.str(); +} /** Formatted output of the arm_compute::MatMulKernelInfo type. * |