24 #ifndef ARM_COMPUTE_CPU_DEPTHWISE_CONV2D_ASSEMBLY_WRAPPER_KERNEL_H 25 #define ARM_COMPUTE_CPU_DEPTHWISE_CONV2D_ASSEMBLY_WRAPPER_KERNEL_H 36 class IDepthwiseCommon;
78 const char *
name()
const override;
88 void pack_parameters(
void *parameters_ptr,
void *bias_ptr,
void *weights_ptr,
size_t ld_weights_col,
size_t ld_weights_row);
94 size_t get_storage_size()
const;
103 size_t get_working_size(
unsigned int num_threads,
unsigned int num_input_channels)
const;
109 bool is_configured()
const;
112 std::unique_ptr<arm_conv::depthwise::IDepthwiseCommon> _kernel_asm;
113 std::vector<int32_t> _multipliers{};
114 std::vector<int32_t> _left_shifts{};
115 std::vector<int32_t> _right_shifts{};
This class is a wrapper for the depthwise convolution assembly kernels.
Common interface for all kernels implemented in C++.
Store the tensor's metadata.
#define ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(TypeName)
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
Information about executing thread and CPU.
template UniqueDepthwiseCommon< float > depthwise(const DepthwiseArgs &, const Nothing &)
im2col_func configure(src_target.info(), dst_target.info(), spatial_kernel, conv_info, has_bias)
Describe a multidimensional execution window.
Status validate(const ITensorInfo *scores_in, const ITensorInfo *boxes_in, const ITensorInfo *batch_splits_in, const ITensorInfo *scores_out, const ITensorInfo *boxes_out, const ITensorInfo *classes, const ITensorInfo *batch_splits_out, const ITensorInfo *keeps, const ITensorInfo *keeps_size, const BoxNMSLimitInfo info)