diff options
Diffstat (limited to 'arm_compute/graph')
-rw-r--r-- | arm_compute/graph/TypeLoader.h | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/arm_compute/graph/TypeLoader.h b/arm_compute/graph/TypeLoader.h index dcdc1736a7..41f382ad1d 100644 --- a/arm_compute/graph/TypeLoader.h +++ b/arm_compute/graph/TypeLoader.h @@ -123,6 +123,30 @@ inline ::std::istream &operator>>(::std::istream &stream, ConvolutionMethod &tar target = Convolution_method_from_name(value); return stream; } + +/** Converts a string to a strong types enumeration @ref DepthwiseConvolutionMethod + * + * @param[in] name String to convert + * + * @return Converted Target enumeration + */ +DepthwiseConvolutionMethod depthwise_convolution_method_from_name(const std::string &name); + +/** Input Stream operator for @ref DepthwiseConvolutionMethod + * + * @param[in] stream Stream to parse + * @param[out] target Output target + * + * @return Updated stream + */ +inline ::std::istream &operator>>(::std::istream &stream, DepthwiseConvolutionMethod &target) +{ + std::string value; + stream >> value; + target = depthwise_convolution_method_from_name(value); + return stream; +} + } // namespace graph } // namespace arm_compute #endif /* __ARM_COMPUTE_GRAPH_TYPE_LOADER_H__ */ |