aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/runtime/CL/functions/CLFlattenLayer.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/runtime/CL/functions/CLFlattenLayer.h')
-rw-r--r--arm_compute/runtime/CL/functions/CLFlattenLayer.h2
1 files changed, 2 insertions, 0 deletions
diff --git a/arm_compute/runtime/CL/functions/CLFlattenLayer.h b/arm_compute/runtime/CL/functions/CLFlattenLayer.h
index b8139c2260..f5f4ff554f 100644
--- a/arm_compute/runtime/CL/functions/CLFlattenLayer.h
+++ b/arm_compute/runtime/CL/functions/CLFlattenLayer.h
@@ -29,7 +29,9 @@
namespace arm_compute
{
+class CLCompileContext;
class ICLTensor;
+class ITensorInfo;
/** Basic function to execute flatten. This function calls the following OpenCL kernel:
*