Skip to content

Commit 362ac42

Browse files
authored
Merge pull request #13134 from tensor-tang/refine/op/prelu
refine prelu
2 parents 6ba2b22 + c441088 commit 362ac42

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

paddle/fluid/operators/prelu_op.h

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,10 +38,9 @@ class PReluKernel : public framework::OpKernel<T> {
3838
auto dim = x->dims();
3939
int index = 0;
4040
int i = 0;
41-
int temp = 0;
4241
if (mode == "channel") {
42+
int temp = numel / (dim[0] * dim[1]);
4343
for (i = 0; i < numel; i++) {
44-
temp = numel / (dim[0] * dim[1]);
4544
index = (i / temp) % dim[1];
4645
o_ptr[i] = x_ptr[i] > 0 ? x_ptr[i] : alpha_ptr[index] * x_ptr[i];
4746
}

0 commit comments

Comments
 (0)