From 6e8c5665b4a2d47ea9de2d347408d0d86d47e50d Mon Sep 17 00:00:00 2001 From: dangqingqing Date: Tue, 6 Jun 2017 16:52:35 +0800 Subject: [PATCH] Add layers into doc. --- doc/api/v2/config/layer.rst | 11 +++++++++++ paddle/function/RowConvOpGpu.cu | 4 ++-- 2 files changed, 13 insertions(+), 2 deletions(-) diff --git a/doc/api/v2/config/layer.rst b/doc/api/v2/config/layer.rst index 1efa74ecda..7c22b67775 100644 --- a/doc/api/v2/config/layer.rst +++ b/doc/api/v2/config/layer.rst @@ -59,6 +59,11 @@ context_projection .. autoclass:: paddle.v2.layer.context_projection :noindex: +row_conv +-------- +.. autoclass:: paddle.v2.layer.row_conv + :noindex: + Image Pooling Layer =================== @@ -346,6 +351,12 @@ sampling_id .. autoclass:: paddle.v2.layer.sampling_id :noindex: +multiplex +--------- +.. autoclass:: paddle.v2.layer.multiplex + :noindex: + + Slicing and Joining Layers ========================== diff --git a/paddle/function/RowConvOpGpu.cu b/paddle/function/RowConvOpGpu.cu index 06e2c2baac..c0b947e224 100644 --- a/paddle/function/RowConvOpGpu.cu +++ b/paddle/function/RowConvOpGpu.cu @@ -312,8 +312,8 @@ void RowConvGrad(const GpuMatrix& outG, dim3 dimBlock(32, 32); dim3 dimGrid(DIVUP(width, dimBlock.x), 1); real* dw = filterG.getData(); - if (contextLength <= 16) { - KeRowConvBwWeight<32, 32, 16> + if (contextLength <= 32) { + KeRowConvBwWeight<32, 32, 32> <<>> (dw, x, dy, starts, height, width, numSeq, contextLength); } else {