[onert-micro] fix wrong gru weight reset (#14093)

chunseoklee · web-flow · commit 50d75f27981d · 2024-09-26T17:24:15.000+09:00
- fix wrong gru reset logic by fixing memset size

ONE-DCO-1.0-Signed-off-by: Chunseok Lee &lt;chunseok.lee@samsung.com&gt;
diff --git a/onert-micro/onert-micro/include/pal/common/PALGRUWeightGrad.h b/onert-micro/onert-micro/include/pal/common/PALGRUWeightGrad.h
@@ -151,8 +151,8 @@ OMStatus GRUWeightGrads(
   two_dim_output_shape.setDim(0, output_shape.dims(dim_count - 2));
   two_dim_output_shape.setDim(1, output_shape.dims(dim_count - 1));
 
-  std::memset(weight_input_grad_data, 0, output_shape.flatSize() * sizeof(float) * time);
-  std::memset(weight_hidden_grad_data, 0, input_shape.dims(2) * sizeof(float) * time);
+  std::memset(weight_input_grad_data, 0, weight_input_shape.flatSize() * sizeof(float));
+  std::memset(weight_hidden_grad_data, 0, weight_hidden_shape.flatSize() * sizeof(float));
 
   for (int i = 0; i < time; ++i)
   {

Original file line number	Diff line number	Diff line change
`@@ -151,8 +151,8 @@ OMStatus GRUWeightGrads(`
`151`	`151`	`two_dim_output_shape.setDim(0, output_shape.dims(dim_count - 2));`
`152`	`152`	`two_dim_output_shape.setDim(1, output_shape.dims(dim_count - 1));`
`153`	`153`
`154`		`- std::memset(weight_input_grad_data, 0, output_shape.flatSize() * sizeof(float) * time);`
`155`		`- std::memset(weight_hidden_grad_data, 0, input_shape.dims(2) * sizeof(float) * time);`
	`154`	`+ std::memset(weight_input_grad_data, 0, weight_input_shape.flatSize() * sizeof(float));`
	`155`	`+ std::memset(weight_hidden_grad_data, 0, weight_hidden_shape.flatSize() * sizeof(float));`
`156`	`156`
`157`	`157`	`for (int i = 0; i < time; ++i)`
`158`	`158`	`{`