Fix IndRNN recurrent scale's gradient shape (TensorFlow).

Issue: #7
lmnt-com · Jun 1, 2020 · 7eaf521 · 7eaf521
1 parent 57eaae8
commit 7eaf521
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/frameworks/tf/indrnn.cc b/frameworks/tf/indrnn.cc
@@ -198,7 +198,7 @@ struct HasteIndrnnGradOp : public OpKernel {
     OP_REQUIRES_OK(context, context->allocate_output(1, dW_shape, &dW));
 
     // Needs to be initialized to 0.
-    const TensorShape du_shape = { hidden_size, hidden_size };
+    const TensorShape du_shape = { hidden_size };
     Tensor* du = nullptr;
     OP_REQUIRES_OK(context, context->allocate_output(2, du_shape, &du));