diff --git a/src/framework/cl/cl_image.h b/src/framework/cl/cl_image.h index a144ccf35b5bf41b3ffbbc65a18b92271a16003c..7c80078ed6909c719e0e064dda8b6b41cd18cf69 100644 --- a/src/framework/cl/cl_image.h +++ b/src/framework/cl/cl_image.h @@ -222,9 +222,9 @@ class CLImage { cl_context context_; }; -void TensorToCLImage(Tensor *tensor, CLImage *image); +void TensorToCLImage(Tensor *tensor, CLImage *image,cl_command_queue commandQueue); -void CLImageToTensor(CLImage *image, Tensor *tensor); +void CLImageToTensor(CLImage *image, Tensor *tensor,cl_command_queue commandQueue); } // namespace framework } // namespace paddle_mobile