Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

fix bug in prelu , issue 12061 #12660

Merged
merged 2 commits into from
Sep 27, 2018
Merged

fix bug in prelu , issue 12061 #12660

Changes from 1 commit
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
71bb481
fix bug in prelu
StOnEGiggity Sep 25, 2018
5241adc
add unit test
StOnEGiggity Sep 26, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion src/operator/leaky_relu-inl.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -301,7 +301,7 @@ class LeakyReLUOp : public Operator {
TShape result(dst.ndim());
int s = src.ndim() - 1;
for (int i = dst.ndim() - 1; i >= 0; i--) {
if (s >= 0 && (dst[i] == src[s] || src[s] == 1)) {
if (s >= 0 && i <= 1 && (dst[i] == src[s] || src[s] == 1)) {
result[i] = src[s];
s--;
} else {
Expand Down