experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update ops-related pbtxt files.

Browse Source 
PiperOrigin-RevId: 334883220
Change-Id: I3f96bed3524ed3257d629a57d914b23a62ae7678
This commit is contained in:
A. Unique TensorFlower 2020-10-01 13:18:20 -07:00 committed by TensorFlower Gardener
parent 3a63cf6b99
commit 4fd2757500
3 changed files with 203 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

96
tensorflow/core/ops/compat/ops_history_v2/FusedBatchNormGradV3.pbtxt
View File

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

103
tensorflow/core/ops/compat/ops_history_v2/FusedBatchNormV3.pbtxt
View File

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

4
tensorflow/core/ops/ops.pbtxt
View File

@ -17475,6 +17475,8 @@ op {
list {
s: "NHWC"
s: "NCHW"
s: "NDHWC"
s: "NCDHW"
}
}
}
@ -17673,6 +17675,8 @@ op {
list {
s: "NHWC"
s: "NCHW"
s: "NDHWC"
s: "NCDHW"
}
}
}