Update ops-related pbtxt files.
PiperOrigin-RevId: 336703857 Change-Id: Ib07e37834b57d1e218063ca242213bb0482d7619
This commit is contained in:
parent
ae6816d52f
commit
594177934c
@ -0,0 +1,79 @@
|
||||
op {
|
||||
name: "QuantizeAndDequantizeV4"
|
||||
input_arg {
|
||||
name: "input"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "input_min"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "input_max"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "T"
|
||||
}
|
||||
attr {
|
||||
name: "signed_input"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: true
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "num_bits"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 8
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "range_given"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: false
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_BFLOAT16
|
||||
type: DT_HALF
|
||||
type: DT_FLOAT
|
||||
type: DT_DOUBLE
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "round_mode"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "HALF_TO_EVEN"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "HALF_TO_EVEN"
|
||||
s: "HALF_UP"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "narrow_range"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: false
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "axis"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: -1
|
||||
}
|
||||
}
|
||||
}
|
@ -0,0 +1,50 @@
|
||||
op {
|
||||
name: "QuantizeAndDequantizeV4Grad"
|
||||
input_arg {
|
||||
name: "gradients"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "input"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "input_min"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "input_max"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "input_backprop"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "input_min_backprop"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "input_max_backprop"
|
||||
type_attr: "T"
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_BFLOAT16
|
||||
type: DT_HALF
|
||||
type: DT_FLOAT
|
||||
type: DT_DOUBLE
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "axis"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: -1
|
||||
}
|
||||
}
|
||||
}
|
@ -29872,6 +29872,135 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "QuantizeAndDequantizeV4"
|
||||
input_arg {
|
||||
name: "input"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "input_min"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "input_max"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "T"
|
||||
}
|
||||
attr {
|
||||
name: "signed_input"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: true
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "num_bits"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 8
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "range_given"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: false
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_BFLOAT16
|
||||
type: DT_HALF
|
||||
type: DT_FLOAT
|
||||
type: DT_DOUBLE
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "round_mode"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "HALF_TO_EVEN"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "HALF_TO_EVEN"
|
||||
s: "HALF_UP"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "narrow_range"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: false
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "axis"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: -1
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "QuantizeAndDequantizeV4Grad"
|
||||
input_arg {
|
||||
name: "gradients"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "input"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "input_min"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "input_max"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "input_backprop"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "input_min_backprop"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "input_max_backprop"
|
||||
type_attr: "T"
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_BFLOAT16
|
||||
type: DT_HALF
|
||||
type: DT_FLOAT
|
||||
type: DT_DOUBLE
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "axis"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: -1
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "QuantizeDownAndShrinkRange"
|
||||
input_arg {
|
||||
|
Loading…
Reference in New Issue
Block a user