File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -116,6 +116,7 @@ extern void hl_maxpool_backward(const int frameCnt,
116
116
* @param[in] paddingW padding width.
117
117
* @param[out] tgtData output data.
118
118
* @param[in] tgtStride stride between output data samples.
119
+ * @param[in] excludeMode whether to consider paddings for size.
119
120
*
120
121
*/
121
122
extern void hl_avgpool_forward (const int frameCnt ,
@@ -132,7 +133,8 @@ extern void hl_avgpool_forward(const int frameCnt,
132
133
const int paddingH ,
133
134
const int paddingW ,
134
135
real * tgtData ,
135
- const int tgtStride );
136
+ const int tgtStride ,
137
+ bool excludeMode );
136
138
137
139
/**
138
140
* @brief Maximum pool backward.
@@ -154,6 +156,7 @@ extern void hl_avgpool_forward(const int frameCnt,
154
156
* @param[in] scaleB scale.
155
157
* @param[out] backGrad output grad.
156
158
* @param[in] outStride stride between output data samples.
159
+ * @param[in] excludeMode whether to consider paddings for size.
157
160
*
158
161
*/
159
162
extern void hl_avgpool_backward (const int frameCnt ,
@@ -172,7 +175,8 @@ extern void hl_avgpool_backward(const int frameCnt,
172
175
real scaleA ,
173
176
real scaleB ,
174
177
real * backGrad ,
175
- const int outStride );
178
+ const int outStride ,
179
+ bool excludeMode );
176
180
177
181
extern void hl_maxpool3D_forward (const int frameCnt ,
178
182
const real * inputData ,
You can’t perform that action at this time.
0 commit comments