From c9e7f42e663b1a5e3432997f6b75848cd9df5721 Mon Sep 17 00:00:00 2001
From: jiangjiajun <jiangjiajun@baidu.com>
Date: Mon, 12 Aug 2019 11:45:02 +0800
Subject: [PATCH] replace pad with pad2d

---
 x2paddle/op_mapper/tf_op_mapper.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/x2paddle/op_mapper/tf_op_mapper.py b/x2paddle/op_mapper/tf_op_mapper.py
index 5140ab7..2136604 100644
--- a/x2paddle/op_mapper/tf_op_mapper.py
+++ b/x2paddle/op_mapper/tf_op_mapper.py
@@ -667,8 +667,14 @@ class TFOpMapper(OpMapper):
         paddings = paddings.value.flatten().tolist()
         if input.tf_data_format == "NHWC" and len(input.out_shapes[0]) == 4:
             paddings = [paddings[i] for i in [0, 1, 6, 7, 2, 3, 4, 5]]
+
+        pad_op = "pad"
+        if len(input.out_shapes[0]) == 4:
+            if paddings[0] + paddings[1] + paddings[2] + paddings[3] == 0:
+                paddings = paddings[4:]
+                pad_op = "pad2d"
         attr = {"paddings": paddings}
-        node.fluid_code.add_layer("pad",
+        node.fluid_code.add_layer(pad_op,
                                   inputs=input,
                                   output=node,
                                   param_attr=attr)
-- 
GitLab