From 0820eeaa893b29d4d45b225b67141dbc5f4b5261 Mon Sep 17 00:00:00 2001 From: Double_V Date: Tue, 8 Jun 2021 15:09:03 +0800 Subject: [PATCH] Op pass pool (#33316) * add pool2d and reshape2 pbtxt * add pool2d and reshape2 pbtxt, test=develop * delete reshape2.txt, test=develop * add lines, test=develop --- paddle/fluid/operators/compat/pool2d.pbtxt | 92 ++++++++++++++++++++ paddle/fluid/operators/compat/reshape2.pbtxt | 53 +++++++++++ 2 files changed, 145 insertions(+) create mode 100644 paddle/fluid/operators/compat/pool2d.pbtxt create mode 100644 paddle/fluid/operators/compat/reshape2.pbtxt diff --git a/paddle/fluid/operators/compat/pool2d.pbtxt b/paddle/fluid/operators/compat/pool2d.pbtxt new file mode 100644 index 00000000000..1620d1ef1c6 --- /dev/null +++ b/paddle/fluid/operators/compat/pool2d.pbtxt @@ -0,0 +1,92 @@ +type: "pool2d" +def { + inputs { + name: "X" + } + outputs { + name: "Out" + } + attrs { + name: "pooling_type" + type: STRING + } + attrs { + name: "ksize" + type: INTS + } + attrs { + name: "global_pooling" + type: BOOLEAN + } + attrs { + name: "strides" + type: INTS + } + attrs { + name: "paddings" + type: INTS + } + attrs { + name: "exclusive" + type: BOOLEAN + } + attrs { + name: "adaptive" + type: BOOLEAN + } + attrs { + name: "ceil_mode" + type: BOOLEAN + } + attrs { + name: "data_format" + type: STRING + } + attrs { + name: "padding_algorithm" + type: STRING + } +} +extra { + attrs { + name: "is_test" + type: BOOLEAN + } + attrs { + name: "use_cudnn" + type: BOOLEAN + } + attrs { + name: "use_mkldnn" + type: BOOLEAN + } + attrs { + name: "use_quantizer" + type: BOOLEAN + } + attrs { + name: "mkldnn_data_type" + type: STRING + } + 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/reshape2.pbtxt b/paddle/fluid/operators/compat/reshape2.pbtxt new file mode 100644 index 00000000000..2ccc83305ba --- /dev/null +++ b/paddle/fluid/operators/compat/reshape2.pbtxt @@ -0,0 +1,53 @@ +type: "reshape2" +def { + inputs { + name: "X" + } + outputs { + name: "Out" + } + attrs { + name: "shape" + type: INTS + } +} +extra { + inputs { + name: "Shape" + } + inputs { + name: "ShapeTensor" + } + outputs { + name: "XShape" + } + attrs { + name: "use_quantizer" + type: BOOLEAN + } + attrs { + name: "mkldnn_data_type" + type: STRING + } + 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