“fea601e92c5709d1ca18313a35d2eaf8a38c8a77”上不存在“mobile/src/operators/kernel/cl/scale_kernel.cpp”
Created by: chengduoZH
fix https://github.com/PaddlePaddle/Paddle/issues/4637
-
cudnn_helper.h
can be used to process 3D data directly. I only addkNCDHW
element and change it's unit test.class DataLayout
are not used in current code. - I only add register code for
pool3d_cudnn_op
and unit test. - Although there are two interface for tensor descriptor in cudnn (
cudnnSetTensorNdDescriptor
andcudnnSetTensor4dDescriptor
), but the current understanding is thatcudnnSetTensorNdDescriptor
can replacecudnnSetTensor4dDescriptor
, and because it is only aDescriptor
, so it is possible that the efficiency of influence is not large (or even negligible).
I'll get a better understanding of the difference between them. If someone know that, please tell me, thx.