Skip to content

Commit e58394a

Browse files
committed
Update THNN.h
1 parent 720aa99 commit e58394a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

lib/THNN/generic/THNN.h

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -291,6 +291,7 @@ TH_API void THNN_(RReLU_updateOutput)(
291291
real upper,
292292
bool train,
293293
bool inplace,
294+
bool channelwise,
294295
THGenerator *generator);
295296
TH_API void THNN_(RReLU_updateGradInput)(
296297
THNNState *state,
@@ -301,7 +302,8 @@ TH_API void THNN_(RReLU_updateGradInput)(
301302
real lower,
302303
real upper,
303304
bool train,
304-
bool inplace);
305+
bool inplace,
306+
bool channelwise);
305307

306308
TH_API void THNN_(Sigmoid_updateOutput)(
307309
THNNState *state,

0 commit comments

Comments
 (0)