fix:paddle._C_ops.layer_norm()

PaddlePaddle · Sep 24, 2024 · e226e78 · e226e78
1 parent 9f3e9b2
commit e226e78
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddleslim/nas/ofa/layers.py b/paddleslim/nas/ofa/layers.py
@@ -1298,7 +1298,7 @@ def forward(self, input):
         self.cur_config = {'prune_dim': feature_dim}
 
         if paddle.in_dynamic_mode():
-            out, _, _ = paddle._C_ops.layer_norm(
+            out = paddle._C_ops.layer_norm(
                 input, weight, bias, self._epsilon, begin_norm_axis, False)
         else:
             paddle.common_ops_import.check_variable_and_dtype(