Skip to content

Commit e9b05c7

Browse files
committed
Use THCTensor rather than THCudaTensor in THCUNN.h definition of
GatedLinearUnit.
1 parent a9785bb commit e9b05c7

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

generic/THCUNN.h

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -140,15 +140,15 @@ TH_API void THNN_(HardTanh_updateGradInput)(
140140

141141
TH_API void THNN_(GatedLinear_updateOutput)(
142142
THCState *state,
143-
THCudaTensor *input,
144-
THCudaTensor *output,
143+
THCTensor *input,
144+
THCTensor *output,
145145
int dim);
146146

147147
TH_API void THNN_(GatedLinear_updateGradInput)(
148148
THCState *state,
149-
THCudaTensor *input,
150-
THCudaTensor *gradOutput,
151-
THCudaTensor *gradInput,
149+
THCTensor *input,
150+
THCTensor *gradOutput,
151+
THCTensor *gradInput,
152152
int dim);
153153

154154
TH_API void THNN_(LeakyReLU_updateOutput)(

0 commit comments

Comments
 (0)