diff options
Diffstat (limited to 'src/gpu/cl/ClContext.h')
| -rw-r--r-- | src/gpu/cl/ClContext.h | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/src/gpu/cl/ClContext.h b/src/gpu/cl/ClContext.h index 2a0d4ee1c8..2c67ccf4d2 100644 --- a/src/gpu/cl/ClContext.h +++ b/src/gpu/cl/ClContext.h @@ -24,11 +24,11 @@ #ifndef SRC_GPU_CLCONTEXT_H #define SRC_GPU_CLCONTEXT_H +#include "arm_compute/core/CL/OpenCL.h" + #include "src/common/IContext.h" #include "src/runtime/CL/mlgo/MLGOHeuristics.h" -#include "arm_compute/core/CL/OpenCL.h" - namespace arm_compute { namespace gpu @@ -74,8 +74,12 @@ public: bool set_cl_ctx(::cl::Context ctx); // Inherrited methods overridden - ITensorV2 *create_tensor(const AclTensorDescriptor &desc, bool allocate) override; - IQueue *create_queue(const AclQueueOptions *options) override; + ITensorV2 *create_tensor(const AclTensorDescriptor &desc, bool allocate) override; + IQueue *create_queue(const AclQueueOptions *options) override; + std::tuple<IOperator *, StatusCode> create_activation(const AclTensorDescriptor &src, + const AclTensorDescriptor &dst, + const AclActivationDescriptor &act, + bool is_validate) override; private: mlgo::MLGOHeuristics _mlgo_heuristics; @@ -86,4 +90,4 @@ private: } // namespace gpu } // namespace arm_compute -#endif /* SRC_GPU_CLCONTEXT_H */
\ No newline at end of file +#endif /* SRC_GPU_CLCONTEXT_H */ |