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

Issue: #7
lmnt-com · Jun 1, 2020 · 51d2a9f · 51d2a9f
1 parent 7eaf521
commit 51d2a9f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/frameworks/tf/layer_norm_indrnn.cc b/frameworks/tf/layer_norm_indrnn.cc
@@ -248,7 +248,7 @@ struct HasteLayerNormIndrnnGradOp : 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));