未验证 提交 c5c37320 编写于 作者: W Wilber 提交者: GitHub

add op_def for gru, lstm and layer_norm (#33317)

* add op_def for gru, lstm and layer_norm

* fix no newline flag
上级 443cf71a
type: "gru"
def {
inputs {
name: "Input"
}
inputs {
name: "H0"
}
inputs {
name: "Weight"
}
inputs {
name: "Bias"
}
outputs {
name: "BatchGate"
}
outputs {
name: "BatchResetHiddenPrev"
}
outputs {
name: "BatchHidden"
}
outputs {
name: "Hidden"
}
attrs {
name: "activation"
type: STRING
}
attrs {
name: "gate_activation"
type: STRING
}
attrs {
name: "is_reverse"
type: BOOLEAN
}
attrs {
name: "origin_mode"
type: BOOLEAN
}
}
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: "layer_norm"
def {
inputs {
name: "X"
}
inputs {
name: "Scale"
}
inputs {
name: "Bias"
}
outputs {
name: "Y"
}
outputs {
name: "Mean"
}
outputs {
name: "Variance"
}
attrs {
name: "epsilon"
type: FLOAT
}
attrs {
name: "begin_norm_axis"
type: INT
}
}
extra {
attrs {
name: "use_mkldnn"
type: BOOLEAN
}
attrs {
name: "mkldnn_data_type"
type: STRING
}
attrs {
name: "is_test"
type: BOOLEAN
}
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: "lstm"
def {
inputs {
name: "Input"
}
inputs {
name: "H0"
}
inputs {
name: "C0"
}
inputs {
name: "Weight"
}
inputs {
name: "Bias"
}
outputs {
name: "Hidden"
}
outputs {
name: "Cell"
}
outputs {
name: "BatchGate"
}
outputs {
name: "BatchCellPreAct"
}
attrs {
name: "use_peepholes"
type: BOOLEAN
}
attrs {
name: "is_reverse"
type: BOOLEAN
}
attrs {
name: "gate_activation"
type: STRING
}
attrs {
name: "cell_activation"
type: STRING
}
attrs {
name: "candidate_activation"
type: STRING
}
}
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.
先完成此消息的编辑!
想要评论请 注册