未验证 提交 0820eeaa 编写于 作者: D Double_V 提交者: GitHub

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
上级 27f4ced8
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
}
}
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
}
}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册