提交 cb5a7a8b 编写于 作者: G guosheng

Fix errors of GPU AvgPooling with the excludeMode argument

上级 192c00a7
...@@ -116,6 +116,7 @@ extern void hl_maxpool_backward(const int frameCnt, ...@@ -116,6 +116,7 @@ extern void hl_maxpool_backward(const int frameCnt,
* @param[in] paddingW padding width. * @param[in] paddingW padding width.
* @param[out] tgtData output data. * @param[out] tgtData output data.
* @param[in] tgtStride stride between output data samples. * @param[in] tgtStride stride between output data samples.
* @param[in] excludeMode whether to consider paddings for size.
* *
*/ */
extern void hl_avgpool_forward(const int frameCnt, extern void hl_avgpool_forward(const int frameCnt,
...@@ -132,7 +133,8 @@ extern void hl_avgpool_forward(const int frameCnt, ...@@ -132,7 +133,8 @@ extern void hl_avgpool_forward(const int frameCnt,
const int paddingH, const int paddingH,
const int paddingW, const int paddingW,
real* tgtData, real* tgtData,
const int tgtStride); const int tgtStride,
bool excludeMode);
/** /**
* @brief Maximum pool backward. * @brief Maximum pool backward.
...@@ -154,6 +156,7 @@ extern void hl_avgpool_forward(const int frameCnt, ...@@ -154,6 +156,7 @@ extern void hl_avgpool_forward(const int frameCnt,
* @param[in] scaleB scale. * @param[in] scaleB scale.
* @param[out] backGrad output grad. * @param[out] backGrad output grad.
* @param[in] outStride stride between output data samples. * @param[in] outStride stride between output data samples.
* @param[in] excludeMode whether to consider paddings for size.
* *
*/ */
extern void hl_avgpool_backward(const int frameCnt, extern void hl_avgpool_backward(const int frameCnt,
...@@ -172,7 +175,8 @@ extern void hl_avgpool_backward(const int frameCnt, ...@@ -172,7 +175,8 @@ extern void hl_avgpool_backward(const int frameCnt,
real scaleA, real scaleA,
real scaleB, real scaleB,
real* backGrad, real* backGrad,
const int outStride); const int outStride,
bool excludeMode);
extern void hl_maxpool3D_forward(const int frameCnt, extern void hl_maxpool3D_forward(const int frameCnt,
const real* inputData, const real* inputData,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册