diff options
Diffstat (limited to 'arm_compute')
| -rw-r--r-- | arm_compute/core/CL/CLHelpers.h | 8 | ||||
| -rw-r--r-- | arm_compute/core/KernelDescriptors.h | 1 |
2 files changed, 5 insertions, 4 deletions
diff --git a/arm_compute/core/CL/CLHelpers.h b/arm_compute/core/CL/CLHelpers.h index edbc705c6f..a162b1c118 100644 --- a/arm_compute/core/CL/CLHelpers.h +++ b/arm_compute/core/CL/CLHelpers.h @@ -242,13 +242,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 +/* Helper function to check if we can export the tensor to cl_image * - * @param[in] tensor Weights tensor + * @param[in] input tensor * - * @return true if we can export the weights to cl_image + * @return true if we can export the tensor to cl_image */ -bool export_weights_to_cl_image(const ITensorInfo *tensor); +bool export_to_cl_image(const ITensorInfo *tensor); /* Helper function to force unroll with pragma when any of the input values (iterations) are greater than @ref max_manual_loop_unrolling * diff --git a/arm_compute/core/KernelDescriptors.h b/arm_compute/core/KernelDescriptors.h index c45be9c06f..cacbef25ea 100644 --- a/arm_compute/core/KernelDescriptors.h +++ b/arm_compute/core/KernelDescriptors.h @@ -106,6 +106,7 @@ struct DWCComputeKernelInfo { unsigned int n0{ 0 }; /**< Number of columns processed by each thread */ unsigned int m0{ 0 }; /**< Number of rows processed by each thread */ + bool export_input_to_cl_image{ false }; /**< Export input to cl_image */ bool export_weights_to_cl_image{ false }; /**< Export the weights to cl_image */ }; |