diff options
Diffstat (limited to 'compute_kernel_writer/src/cl/CLTensorArgument.h')
| -rw-r--r-- | compute_kernel_writer/src/cl/CLTensorArgument.h | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/compute_kernel_writer/src/cl/CLTensorArgument.h b/compute_kernel_writer/src/cl/CLTensorArgument.h index 35df51422e..a79cf340bb 100644 --- a/compute_kernel_writer/src/cl/CLTensorArgument.h +++ b/compute_kernel_writer/src/cl/CLTensorArgument.h @@ -27,6 +27,7 @@ #include "ckw/types/TensorComponentType.h" #include "ckw/types/TensorStorageType.h" +#include "src/cl/CLTensorComponent.h" #include "src/ITensor.h" #include <memory> @@ -39,8 +40,6 @@ namespace ckw class TensorInfo; class ITensorComponent; -class CLTensorComponent; -class CLTensorStorage; /** OpenCL specific tensor argument * Internally, the object keeps track of the components and storages used to minimize the number |