From 7cadd957b63d4a833accb4b66ce0022f34ecddc8 Mon Sep 17 00:00:00 2001 From: cuicheng01 <45199522+cuicheng01@users.noreply.github.com> Date: Tue, 8 Jun 2021 15:49:03 +0800 Subject: [PATCH] add squeeze2 and unsqueeze2 pbtxt (#33343) * Create squeeze2.pbtxt * Create unsqueeze2.pbtxt * Update unsqueeze2.pbtxt * Update squeeze2.pbtxt --- paddle/fluid/operators/compat/squeeze2.pbtxt | 38 ++++++++++++++++ .../fluid/operators/compat/unsqueeze2.pbtxt | 44 +++++++++++++++++++ 2 files changed, 82 insertions(+) create mode 100644 paddle/fluid/operators/compat/squeeze2.pbtxt create mode 100644 paddle/fluid/operators/compat/unsqueeze2.pbtxt diff --git a/paddle/fluid/operators/compat/squeeze2.pbtxt b/paddle/fluid/operators/compat/squeeze2.pbtxt new file mode 100644 index 00000000000..160e6a72786 --- /dev/null +++ b/paddle/fluid/operators/compat/squeeze2.pbtxt @@ -0,0 +1,38 @@ +type: "squeeze2" +def { + inputs { + name: "X" + } + outputs { + name: "Out" + } + outputs { + name: "XShape" + } + attrs { + name: "axes" + type: INTS + } +} +extra { + attrs { + name: "op_role" + type: INT + } + attrs { + name: "op_role_var" + type: STRINGS + } + attrs { + name: "op_namescope" + type: STRING + } + attrs { + name: "op_callstack" + type: STRINGS + } + attrs { + name: "op_device" + type: STRING + } +} diff --git a/paddle/fluid/operators/compat/unsqueeze2.pbtxt b/paddle/fluid/operators/compat/unsqueeze2.pbtxt new file mode 100644 index 00000000000..ed3c32754a5 --- /dev/null +++ b/paddle/fluid/operators/compat/unsqueeze2.pbtxt @@ -0,0 +1,44 @@ +type: "unsqueeze2" +def { + inputs { + name: "X" + } + inputs { + name: "AxesTensor" + } + inputs { + name: "AxesTensorList" + } + outputs { + name: "Out" + } + outputs { + name: "XShape" + } + attrs { + name: "axes" + type: INTS + } +} +extra { + attrs { + name: "op_role" + type: INT + } + attrs { + name: "op_role_var" + type: STRINGS + } + attrs { + name: "op_namescope" + type: STRING + } + attrs { + name: "op_callstack" + type: STRINGS + } + attrs { + name: "op_device" + type: STRING + } +} -- GitLab