未验证 提交 7cadd957 编写于 作者: C cuicheng01 提交者: GitHub

add squeeze2 and unsqueeze2 pbtxt (#33343)

* Create squeeze2.pbtxt

* Create unsqueeze2.pbtxt

* Update unsqueeze2.pbtxt

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