diff options
Diffstat (limited to 'arm_compute/core/CL')
| -rw-r--r-- | arm_compute/core/CL/CLHelpers.h | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/arm_compute/core/CL/CLHelpers.h b/arm_compute/core/CL/CLHelpers.h index a9ac6a5933..180211c558 100644 --- a/arm_compute/core/CL/CLHelpers.h +++ b/arm_compute/core/CL/CLHelpers.h @@ -236,5 +236,13 @@ 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 */ |