aboutsummaryrefslogtreecommitdiff
path: root/src/core/CL/CLUtils.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/core/CL/CLUtils.h')
-rw-r--r--src/core/CL/CLUtils.h10
1 files changed, 10 insertions, 0 deletions
diff --git a/src/core/CL/CLUtils.h b/src/core/CL/CLUtils.h
index b31944c72f..35dbee723e 100644
--- a/src/core/CL/CLUtils.h
+++ b/src/core/CL/CLUtils.h
@@ -33,6 +33,7 @@ namespace arm_compute
class TensorShape;
class CLBuildOptions;
class ITensorInfo;
+class ICLTensor;
/** OpenCL Image2D types */
enum class CLImage2DType
@@ -41,6 +42,15 @@ enum class CLImage2DType
WriteOnly
};
+/** Create a cl::Image2D object from a tensor
+ *
+ * @param[in] tensor Tensor from which to construct Image 2D object
+ * @param[in] image_type Image 2D type (@ref CLImage2DType)
+ *
+ * @return cl::Image2D object
+ */
+cl::Image2D create_image2d_from_tensor(const ICLTensor *tensor, CLImage2DType image_type);
+
/** Create a cl::Image2D object from an OpenCL buffer
*
* @note The following conditions are required to create a OpenCL image object from OpenCL buffer,