未验证 提交 2c10ca64 编写于 作者: C cc 提交者: GitHub

Add op def for quant ops (#33351)

上级 d46b4086
type: "fake_channel_wise_quantize_abs_max"
def {
inputs {
name: "X"
}
outputs {
name: "Out"
}
outputs {
name: "OutScale"
}
attrs {
name: "quant_axis"
type: INT
}
attrs {
name: "bit_length"
type: INT
}
}
extra {
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: "fake_dequantize_max_abs"
def {
inputs {
name: "X"
}
inputs {
name: "Scale"
}
outputs {
name: "Out"
}
attrs {
name: "max_range"
type: FLOAT
}
}
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: "fake_quantize_abs_max"
def {
inputs {
name: "X"
}
outputs {
name: "Out"
}
outputs {
name: "OutScale"
}
attrs {
name: "bit_length"
type: INT
}
}
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: "fake_quantize_moving_average_abs_max"
def {
inputs {
name: "X"
}
inputs {
name: "InScale"
}
inputs {
name: "InAccum"
}
inputs {
name: "InState"
}
outputs {
name: "Out"
}
outputs {
name: "OutScale"
}
outputs {
name: "OutState"
}
outputs {
name: "OutAccum"
}
attrs {
name: "moving_rate"
type: FLOAT
}
attrs {
name: "bit_length"
type: INT
}
}
extra {
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: "fake_quantize_range_abs_max"
def {
inputs {
name: "X"
}
inputs {
name: "InScale"
}
inputs {
name: "Iter"
}
outputs {
name: "Out"
}
outputs {
name: "OutScale"
}
outputs {
name: "OutScales"
}
attrs {
name: "window_size"
type: INT
}
attrs {
name: "bit_length"
type: INT
}
}
extra {
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: "fill_constant"
def {
inputs {
name: "ValueTensor"
}
inputs {
name: "ShapeTensor"
}
inputs {
name: "ShapeTensorList"
}
outputs {
name: "Out"
}
attrs {
name: "dtype"
type: INT
}
attrs {
name: "shape"
type: LONGS
}
attrs {
name: "value"
type: FLOAT
}
attrs {
name: "str_value"
type: STRING
}
}
extra {
attrs {
name: "force_cpu"
type: BOOLEAN
}
attrs {
name: "place_type"
type: INT
}
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.
先完成此消息的编辑!
想要评论请 注册