We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8095147 commit c36f884Copy full SHA for c36f884
blocksparse/ewops.py
@@ -159,9 +159,11 @@ def ew_z_xb_grad(op, dz):
159
lstm_gates4_op = _op_module.lstm_gates4
160
lstm_gates4_grad_op = _op_module.lstm_gates4_grad
161
162
-def fused_lstm_gates(c, *args, name=None):
+def fused_lstm_gates(c, *args, **kwargs):
163
# returns c_next, h_next
164
-
+
165
+ assert len(kwargs) <= 1
166
+ name = kwargs.pop('name', None)
167
# args is h (all four gates fused in single tensor)
168
if len(args) == 1:
169
return lstm_gates_op(c, args[0], name=name)
0 commit comments