diff --git a/x2paddle/op_mapper/static/tf2paddle/tf_op_mapper.py b/x2paddle/op_mapper/static/tf2paddle/tf_op_mapper.py
index f037dab6321623354ff0ce5c2f98605ee26aabd4..251e6cdcf0e6b853c3bcb89bdec82fde41ca71cb 100644
--- a/x2paddle/op_mapper/static/tf2paddle/tf_op_mapper.py
+++ b/x2paddle/op_mapper/static/tf2paddle/tf_op_mapper.py
@@ -746,17 +746,13 @@ class TFOpMapper(OpMapper):
         input = self.graph.get_input_node(node, 0)
         input_name = input.name
         self.paddle_graph.add_layer(
-            kernel="fluid.layers.size",
+            kernel="paddle.shape",
             inputs={"input": input_name},
             outputs=[node.name])
-#         self.paddle_graph.add_layer(
-#             kernel="paddle.shape",
-#             inputs={"input": input_name},
-#             outputs=[node.name])
-#         self.paddle_graph.add_layer(
-#             kernel="paddle.prod",
-#             inputs={"x": node.name},
-#             outputs=[node.name])
+        self.paddle_graph.add_layer(
+            kernel="paddle.prod",
+            inputs={"x": node.name},
+            outputs=[node.name])
         
     def Ceil(self, node):
         input = self.graph.get_input_node(node, 0)