fix bug in UT

Author: zhengqiwen1997

python/paddle/fluid/tests/unittests/ir/test_ir_embedding_eltwise_layernorm_fuse_pass.py

@@ -54,7 +54,7 @@ def setUp(self):
             )
             add1 = paddle.add(word_emb, pos_emb)
             add2 = paddle.add(add1, sent_emb)
-            hidden1 = fluid.layers.layer_norm(input=add2, begin_norm_axis=2)
+            hidden1 = paddle.static.nn.layer_norm(input=add2, begin_norm_axis=2)
 
             id1 = fluid.layers.data(
                 name="id1",
@@ -95,7 +95,9 @@ def setUp(self):
             add_1 = paddle.add(emb1, emb2)
             add_2 = paddle.add(add_1, emb3)
             add_3 = paddle.add(add_2, emb4)
-            hidden_1 = fluid.layers.layer_norm(input=add_3, begin_norm_axis=2)
+            hidden_1 = paddle.static.nn.layer_norm(
+                input=add_3, begin_norm_axis=2
+            )
 
         self.feeds = {
             "word_id": np.random.randint(

python/paddle/fluid/tests/unittests/ir/test_ir_skip_layernorm_pass.py

@@ -32,7 +32,7 @@ def setUp(self):
                 name="y", shape=[128, 768], dtype="float32", lod_level=0
             )
             elementwise_out = paddle.add(x=x, y=y)
-            out = fluid.layers.layer_norm(input=elementwise_out)
+            out = paddle.static.nn.layer_norm(input=elementwise_out)
 
         self.fetch_list = [out]
         self.pass_names = "skip_layernorm_fuse_pass"