add tf SplitV

x2paddle/op_mapper/dygraph/tf2paddle/tf_op_mapper.py

@@ -1066,6 +1066,24 @@ class TFOpMapper(OpMapper):
             num_or_sections=num_split,
             axis=dim)
         
+    def SplitV(self, node):
+        input = self.graph.get_input_node(node, 0)
+        size_splits = self.graph.get_input_node(node, 1)
+        assert size_splits.layer_type == "Const", "size_splits of SplitV OP should be Const"
+        size_splits = size_splits.value.tolist()
+        dim = self.graph.get_input_node(node, 2)
+        assert dim.layer_type == "Const", "dim of SplitV OP should be Const"
+        dim = dim.value
+        
+        self.paddle_graph.add_layer(
+            kernel="paddle.split",
+            inputs={"x": input.name},
+            outputs=[
+                "{}_p{}".format(node.layer_name, i) for i in range(len(size_splits))
+            ],
+            num_or_sections=size_splits,
+            axis=dim)
+
     def Slice(self, node):
         input = self.graph.get_input_node(node, 0)
         begin = self.graph.get_input_node(node, 1)

x2paddle/op_mapper/static/tf2paddle/tf_op_mapper.py

@@ -1043,6 +1043,24 @@ class TFOpMapper(OpMapper):
             num_or_sections=num_split,
             axis=dim)
         
+    def SplitV(self, node):
+        input = self.graph.get_input_node(node, 0)
+        size_splits = self.graph.get_input_node(node, 1)
+        assert size_splits.layer_type == "Const", "size_splits of SplitV OP should be Const"
+        size_splits = size_splits.value.tolist()
+        dim = self.graph.get_input_node(node, 2)
+        assert dim.layer_type == "Const", "dim of SplitV OP should be Const"
+        dim = dim.value
+        
+        self.paddle_graph.add_layer(
+            kernel="paddle.split",
+            inputs={"x": input.name},
+            outputs=[
+                "{}_p{}".format(node.layer_name, i) for i in range(len(size_splits))
+            ],
+            num_or_sections=size_splits,
+            axis=dim)
+
     def Slice(self, node):
         input = self.graph.get_input_node(node, 0)
         begin = self.graph.get_input_node(node, 1)