diff options
Diffstat (limited to 'arm_compute/core')
| -rw-r--r-- | arm_compute/core/CL/CLHelpers.h | 8 | ||||
| -rw-r--r-- | arm_compute/core/KernelDescriptors.h | 14 |
2 files changed, 9 insertions, 13 deletions
diff --git a/arm_compute/core/CL/CLHelpers.h b/arm_compute/core/CL/CLHelpers.h index 180211c558..a9ac6a5933 100644 --- a/arm_compute/core/CL/CLHelpers.h +++ b/arm_compute/core/CL/CLHelpers.h @@ -236,13 +236,5 @@ bool get_wbsm_support_info(const cl::Device &device); */ void set_wbsm(cl::Kernel &kernel, cl_int wbsm_hint); -/* Helper function to check if we can export the weights to cl_image - * - * @param[in] tensor Weights tensor - * - * @return true if we can export the weights to cl_image - */ -bool export_weights_to_cl_image(const ITensorInfo *tensor); - } // namespace arm_compute #endif /* ARM_COMPUTE_CLHELPERS_H */ diff --git a/arm_compute/core/KernelDescriptors.h b/arm_compute/core/KernelDescriptors.h index a6e5c3372e..6c1fc74b1e 100644 --- a/arm_compute/core/KernelDescriptors.h +++ b/arm_compute/core/KernelDescriptors.h @@ -96,12 +96,16 @@ struct GEMMKernelInfo GEMMLowpOutputStageInfo output_stage{}; /**< GEMMLowp output stage information */ }; -/** Compute descriptor used by the depthwise convolution native kernel */ -struct DWCComputeKernelInfo +/** Descriptor used by the depthwise convolution kernels */ +struct DWCKernelInfo { - unsigned int n0{ 0 }; /**< Number of columns processed by each thread */ - unsigned int m0{ 0 }; /**< Number of rows processed by each thread */ - bool export_weights_to_cl_image{ false }; /**< Export the weights to cl_image */ + ActivationLayerInfo activation_info{}; /**< Activation function to perform after the depthwise convolution */ +}; + +/** Descriptor used by the depthwise convolution kernels to retrieve the number of output elements processed by each thread */ +struct DWCWeightsKernelInfo +{ + unsigned int n0{ 0 }; /**< Number of columns processed by each thread */ }; /** Descriptor used by the softmax kernels */ |