24 #ifndef ARM_COMPUTE_CPU_DEPTHWISE_CONV2D_H
25 #define ARM_COMPUTE_CPU_DEPTHWISE_CONV2D_H
109 class CpuDepthwiseConv2dOptimizedInternal :
public ICpuOperator
113 CpuDepthwiseConv2dOptimizedInternal() =
default;
115 CpuDepthwiseConv2dOptimizedInternal(
const CpuDepthwiseConv2dOptimizedInternal &) =
delete;
117 CpuDepthwiseConv2dOptimizedInternal(CpuDepthwiseConv2dOptimizedInternal &&) =
default;
119 CpuDepthwiseConv2dOptimizedInternal &
operator=(
const CpuDepthwiseConv2dOptimizedInternal &) =
delete;
121 CpuDepthwiseConv2dOptimizedInternal &
operator=(CpuDepthwiseConv2dOptimizedInternal &&) =
default;
123 ~CpuDepthwiseConv2dOptimizedInternal() =
default;
155 std::unique_ptr<CpuDepthwiseConv2dAssemblyDispatch> _dwc_optimized_func{
nullptr};
156 std::unique_ptr<CpuPermute> _permute_input{
nullptr};
157 std::unique_ptr<CpuPermute> _permute_weights{
nullptr};
158 std::unique_ptr<CpuPermute> _permute_output{
nullptr};
159 std::unique_ptr<CpuActivation> _activationlayer_function{
nullptr};
160 bool _has_bias{
false};
161 bool _is_quantized{
false};
163 bool _permute{
false};
164 bool _is_activationlayer_enabled{
false};
165 bool _is_prepared{
false};
166 bool _are_weights_const{
true};
178 CpuDepthwiseConv2dGeneric() =
default;
180 CpuDepthwiseConv2dGeneric(
const CpuDepthwiseConv2dGeneric &) =
delete;
182 CpuDepthwiseConv2dGeneric(CpuDepthwiseConv2dGeneric &&) =
default;
184 CpuDepthwiseConv2dGeneric &
operator=(
const CpuDepthwiseConv2dGeneric &) =
delete;
186 CpuDepthwiseConv2dGeneric &
operator=(CpuDepthwiseConv2dGeneric &&) =
default;
188 ~CpuDepthwiseConv2dGeneric() =
default;
222 std::unique_ptr<kernels::CpuDepthwiseConv2dNativeKernel> _depthwise_conv_kernel{
nullptr};
223 std::unique_ptr<CpuPermute> _permute_input{
nullptr};
224 std::unique_ptr<CpuPermute> _permute_weights{
nullptr};
225 std::unique_ptr<CpuPermute> _permute_output{
nullptr};
226 std::unique_ptr<CpuActivation> _activationlayer_function{
nullptr};
228 bool _is_prepared{
false};
229 bool _is_activationlayer_enabled{
false};
233 CpuDepthwiseConv2dOptimizedInternal _func_optimized{};
234 CpuDepthwiseConv2dGeneric _func_generic{};