Update ops-related pbtxt files.
PiperOrigin-RevId: 334883220 Change-Id: I3f96bed3524ed3257d629a57d914b23a62ae7678
This commit is contained in:
parent
3a63cf6b99
commit
4fd2757500
@ -92,3 +92,99 @@ op {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
op {
|
||||||
|
name: "FusedBatchNormGradV3"
|
||||||
|
input_arg {
|
||||||
|
name: "y_backprop"
|
||||||
|
type_attr: "T"
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "x"
|
||||||
|
type_attr: "T"
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "scale"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "reserve_space_1"
|
||||||
|
type_attr: "U"
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "reserve_space_2"
|
||||||
|
type_attr: "U"
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "reserve_space_3"
|
||||||
|
type_attr: "U"
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "x_backprop"
|
||||||
|
type_attr: "T"
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "scale_backprop"
|
||||||
|
type_attr: "U"
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "offset_backprop"
|
||||||
|
type_attr: "U"
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "reserve_space_4"
|
||||||
|
type_attr: "U"
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "reserve_space_5"
|
||||||
|
type_attr: "U"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "T"
|
||||||
|
type: "type"
|
||||||
|
allowed_values {
|
||||||
|
list {
|
||||||
|
type: DT_HALF
|
||||||
|
type: DT_BFLOAT16
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "U"
|
||||||
|
type: "type"
|
||||||
|
allowed_values {
|
||||||
|
list {
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "epsilon"
|
||||||
|
type: "float"
|
||||||
|
default_value {
|
||||||
|
f: 0.0001
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "data_format"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: "NHWC"
|
||||||
|
}
|
||||||
|
allowed_values {
|
||||||
|
list {
|
||||||
|
s: "NHWC"
|
||||||
|
s: "NCHW"
|
||||||
|
s: "NDHWC"
|
||||||
|
s: "NCDHW"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "is_training"
|
||||||
|
type: "bool"
|
||||||
|
default_value {
|
||||||
|
b: true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@ -99,3 +99,106 @@ op {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
op {
|
||||||
|
name: "FusedBatchNormV3"
|
||||||
|
input_arg {
|
||||||
|
name: "x"
|
||||||
|
type_attr: "T"
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "scale"
|
||||||
|
type_attr: "U"
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "offset"
|
||||||
|
type_attr: "U"
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "mean"
|
||||||
|
type_attr: "U"
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "variance"
|
||||||
|
type_attr: "U"
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "y"
|
||||||
|
type_attr: "T"
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "batch_mean"
|
||||||
|
type_attr: "U"
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "batch_variance"
|
||||||
|
type_attr: "U"
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "reserve_space_1"
|
||||||
|
type_attr: "U"
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "reserve_space_2"
|
||||||
|
type_attr: "U"
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "reserve_space_3"
|
||||||
|
type_attr: "U"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "T"
|
||||||
|
type: "type"
|
||||||
|
allowed_values {
|
||||||
|
list {
|
||||||
|
type: DT_HALF
|
||||||
|
type: DT_BFLOAT16
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "U"
|
||||||
|
type: "type"
|
||||||
|
allowed_values {
|
||||||
|
list {
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "epsilon"
|
||||||
|
type: "float"
|
||||||
|
default_value {
|
||||||
|
f: 0.0001
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "exponential_avg_factor"
|
||||||
|
type: "float"
|
||||||
|
default_value {
|
||||||
|
f: 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "data_format"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: "NHWC"
|
||||||
|
}
|
||||||
|
allowed_values {
|
||||||
|
list {
|
||||||
|
s: "NHWC"
|
||||||
|
s: "NCHW"
|
||||||
|
s: "NDHWC"
|
||||||
|
s: "NCDHW"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "is_training"
|
||||||
|
type: "bool"
|
||||||
|
default_value {
|
||||||
|
b: true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@ -17475,6 +17475,8 @@ op {
|
|||||||
list {
|
list {
|
||||||
s: "NHWC"
|
s: "NHWC"
|
||||||
s: "NCHW"
|
s: "NCHW"
|
||||||
|
s: "NDHWC"
|
||||||
|
s: "NCDHW"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -17673,6 +17675,8 @@ op {
|
|||||||
list {
|
list {
|
||||||
s: "NHWC"
|
s: "NHWC"
|
||||||
s: "NCHW"
|
s: "NCHW"
|
||||||
|
s: "NDHWC"
|
||||||
|
s: "NCDHW"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user