Update ops-related pbtxt files.

PiperOrigin-RevId: 336703857
Change-Id: Ib07e37834b57d1e218063ca242213bb0482d7619
This commit is contained in:
A. Unique TensorFlower 2020-10-12 11:18:37 -07:00 committed by TensorFlower Gardener
parent ae6816d52f
commit 594177934c
3 changed files with 258 additions and 0 deletions

View File

@ -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
}
}
}

View File

@ -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
}
}
}

View File

@ -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 {