diff --git a/paddle/function/DepthwiseConvOp.cpp b/paddle/function/DepthwiseConvOp.cpp index 358135e9a1..31eccda67d 100644 --- a/paddle/function/DepthwiseConvOp.cpp +++ b/paddle/function/DepthwiseConvOp.cpp @@ -60,6 +60,7 @@ public: int paddingH, int paddingW, T* inputGrad) {} + // TODO(zhaolong) : cpu implementation of depthwise convolution }; template @@ -82,6 +83,7 @@ public: int paddingW, T* colData, T* filterGrad) {} + // TODO(zhaolong) : cpu implementation of depthwise convolution }; /* diff --git a/paddle/function/DepthwiseConvOp.h b/paddle/function/DepthwiseConvOp.h index 5c5a70e5df..356ff37c6a 100644 --- a/paddle/function/DepthwiseConvOp.h +++ b/paddle/function/DepthwiseConvOp.h @@ -122,11 +122,8 @@ public: * \param[in] paddingH padding size in height direction. * \param[in] paddingW padding size in width direction. * \param[in] colData Auxiliary data when calculating filterGrad. - * size: - *inputChannels*filterHeight*filterWidth*outputHeight*outputWidth \param[in] - *multiplierData Auxiliary data when calculating filterGrad. size: - *outputHeight * outputWidth. \param[out] - *filterGrad the grad data of filter. + * \param[in] multiplierData Auxiliary data when calculating filterGrad. + * \param[out] filterGrad the grad data of filter. * */ template diff --git a/paddle/function/DepthwiseConvOpGpu.cu b/paddle/function/DepthwiseConvOpGpu.cu index 5fb85df489..737f091ab8 100644 --- a/paddle/function/DepthwiseConvOpGpu.cu +++ b/paddle/function/DepthwiseConvOpGpu.cu @@ -17,6 +17,7 @@ limitations under the License. */ #include "paddle/math/BaseMatrix.h" namespace paddle { + // CUDA kernel to compute the depthwise convolution forward pass template __global__ diff --git a/paddle/gserver/layers/DepthwiseConvLayer.h b/paddle/gserver/layers/DepthwiseConvLayer.h index ce074803ab..1b154bd99d 100644 --- a/paddle/gserver/layers/DepthwiseConvLayer.h +++ b/paddle/gserver/layers/DepthwiseConvLayer.h @@ -22,7 +22,7 @@ namespace paddle { /** * @brief A subclass of convolution layer. - * This layer do the depthwise convolution calculation in mobilenet. + * This layer does the depthwise convolution calculation of mobilenet. * The config file api is img_depthwise_conv_layer. */