diff options
Diffstat (limited to 'arm_compute/function_info/ConvolutionInfo.h')
| -rw-r--r-- | arm_compute/function_info/ConvolutionInfo.h | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/arm_compute/function_info/ConvolutionInfo.h b/arm_compute/function_info/ConvolutionInfo.h index c27dc523c8..4830cae137 100644 --- a/arm_compute/function_info/ConvolutionInfo.h +++ b/arm_compute/function_info/ConvolutionInfo.h @@ -33,14 +33,18 @@ namespace arm_compute struct ConvolutionInfo { ConvolutionInfo() = default; - ConvolutionInfo(const PadStrideInfo &pad_stride_info, unsigned int depth_multiplier, const ActivationLayerInfo &act_info, const Size2D &dilation) + ConvolutionInfo(const PadStrideInfo &pad_stride_info, + unsigned int depth_multiplier, + const ActivationLayerInfo &act_info, + const Size2D &dilation) : pad_stride_info(pad_stride_info), depth_multiplier(depth_multiplier), act_info(act_info), dilation(dilation) { } - PadStrideInfo pad_stride_info{}; /**< Convolution info (Pads, strides,...) */ - unsigned int depth_multiplier{ 1 }; /**< Multiplier to apply to input's depth to retrieve the output depth. Defaults to 1 */ - ActivationLayerInfo act_info{}; /**< Fused activation to apply after convolution. */ - Size2D dilation{ Size2D(1, 1) }; /**< Dilation, in elements, across x and y. Defaults to (1, 1). */ + PadStrideInfo pad_stride_info{}; /**< Convolution info (Pads, strides,...) */ + unsigned int depth_multiplier{ + 1}; /**< Multiplier to apply to input's depth to retrieve the output depth. Defaults to 1 */ + ActivationLayerInfo act_info{}; /**< Fused activation to apply after convolution. */ + Size2D dilation{Size2D(1, 1)}; /**< Dilation, in elements, across x and y. Defaults to (1, 1). */ }; } // namespace arm_compute #endif /* ACL_ARM_COMPUTE_FUNCTION_INFO_CONVOLUTIONINFO */ |