diff options
Diffstat (limited to 'arm_compute/graph/CL/CLUnmap.h')
-rw-r--r-- | arm_compute/graph/CL/CLUnmap.h | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/arm_compute/graph/CL/CLUnmap.h b/arm_compute/graph/CL/CLUnmap.h index a72706353b..17745c436b 100644 --- a/arm_compute/graph/CL/CLUnmap.h +++ b/arm_compute/graph/CL/CLUnmap.h @@ -29,11 +29,11 @@ namespace arm_compute { -class CLTensor; +class ICLTensor; namespace graph { -class Tensor; +class ITensorObject; /** OpenCL un-map function */ class CLUnmap : public arm_compute::IFunction { @@ -42,7 +42,7 @@ public: * * @param[in] tensor Tensor to un-map */ - CLUnmap(Tensor *tensor); + CLUnmap(ITensorObject *tensor); /** Prevent instances from being copy constructed */ CLUnmap(const CLUnmap &) = delete; /** Prevent instances from being copy assigned */ @@ -56,7 +56,7 @@ public: void run() override; private: - arm_compute::CLTensor *_tensor; /**< Tensor */ + arm_compute::ICLTensor *_tensor; /**< Tensor */ }; } // namespace graph } // namespace arm_compute |