Skip to content

Commit e226e78

Browse files
Ji XingJi Xing
Ji Xing
authored and
Ji Xing
committed
fix:paddle._C_ops.layer_norm()
1 parent 9f3e9b2 commit e226e78

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddleslim/nas/ofa/layers.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1298,7 +1298,7 @@ def forward(self, input):
12981298
self.cur_config = {'prune_dim': feature_dim}
12991299

13001300
if paddle.in_dynamic_mode():
1301-
out, _, _ = paddle._C_ops.layer_norm(
1301+
out = paddle._C_ops.layer_norm(
13021302
input, weight, bias, self._epsilon, begin_norm_axis, False)
13031303
else:
13041304
paddle.common_ops_import.check_variable_and_dtype(

0 commit comments

Comments
 (0)