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

Update ops-related pbtxt files.

Browse Source 
PiperOrigin-RevId: 338326619
Change-Id: I88e8e1d307997485a0392294beaa78fa9bf9dd0d
This commit is contained in:
A. Unique TensorFlower 2020-10-21 13:18:31 -07:00 committed by TensorFlower Gardener
parent 38b03d8d05
commit b996deed07
5 changed files with 368 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

68
tensorflow/core/ops/compat/ops_history_v2/CollectiveGather.pbtxt
View File

@ -142,3 +142,71 @@ op {
}
is_stateful: true
}
op {
name: "CollectiveGather"
input_arg {
name: "input"
type_attr: "T"
}
input_arg {
name: "ordering_token"
type: DT_RESOURCE
number_attr: "Nordering_token"
}
output_arg {
name: "data"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_HALF
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "group_size"
type: "int"
}
attr {
name: "group_key"
type: "int"
}
attr {
name: "instance_key"
type: "int"
}
attr {
name: "shape"
type: "shape"
}
attr {
name: "communication_hint"
type: "string"
default_value {
s: "auto"
}
}
attr {
name: "timeout_seconds"
type: "float"
default_value {
f: 0
}
}
attr {
name: "Nordering_token"
type: "int"
default_value {
i: 0
}
has_minimum: true
}
is_stateful: true
}

64
tensorflow/core/ops/compat/ops_history_v2/CollectiveGatherV2.pbtxt
View File

@ -49,3 +49,67 @@ op {
}
is_stateful: true
}
op {
name: "CollectiveGatherV2"
input_arg {
name: "input"
type_attr: "T"
}
input_arg {
name: "group_size"
type: DT_INT32
}
input_arg {
name: "group_key"
type: DT_INT32
}
input_arg {
name: "instance_key"
type: DT_INT32
}
input_arg {
name: "ordering_token"
type: DT_RESOURCE
number_attr: "Nordering_token"
}
output_arg {
name: "data"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_HALF
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "communication_hint"
type: "string"
default_value {
s: "auto"
}
}
attr {
name: "timeout_seconds"
type: "float"
default_value {
f: 0
}
}
attr {
name: "Nordering_token"
type: "int"
default_value {
i: 0
}
has_minimum: true
}
is_stateful: true
}

98
tensorflow/core/ops/compat/ops_history_v2/CollectiveReduce.pbtxt
View File

@ -295,3 +295,101 @@ op {
}
is_stateful: true
}
op {
name: "CollectiveReduce"
input_arg {
name: "input"
type_attr: "T"
}
input_arg {
name: "ordering_token"
type: DT_RESOURCE
number_attr: "Nordering_token"
}
output_arg {
name: "data"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_HALF
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "group_size"
type: "int"
}
attr {
name: "group_key"
type: "int"
}
attr {
name: "instance_key"
type: "int"
}
attr {
name: "merge_op"
type: "string"
allowed_values {
list {
s: "Min"
s: "Max"
s: "Mul"
s: "Add"
}
}
}
attr {
name: "final_op"
type: "string"
allowed_values {
list {
s: "Id"
s: "Div"
}
}
}
attr {
name: "subdiv_offsets"
type: "list(int)"
}
attr {
name: "wait_for"
type: "list(int)"
default_value {
list {
}
}
}
attr {
name: "communication_hint"
type: "string"
default_value {
s: "auto"
}
}
attr {
name: "timeout_seconds"
type: "float"
default_value {
f: 0
}
}
attr {
name: "Nordering_token"
type: "int"
default_value {
i: 0
}
has_minimum: true
}
is_stateful: true
}

86
tensorflow/core/ops/compat/ops_history_v2/CollectiveReduceV2.pbtxt
View File

@ -137,3 +137,89 @@ op {
}
is_stateful: true
}
op {
name: "CollectiveReduceV2"
input_arg {
name: "input"
type_attr: "T"
}
input_arg {
name: "group_size"
type: DT_INT32
}
input_arg {
name: "group_key"
type: DT_INT32
}
input_arg {
name: "instance_key"
type: DT_INT32
}
input_arg {
name: "ordering_token"
type: DT_RESOURCE
number_attr: "Nordering_token"
}
output_arg {
name: "data"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_HALF
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "merge_op"
type: "string"
allowed_values {
list {
s: "Min"
s: "Max"
s: "Mul"
s: "Add"
}
}
}
attr {
name: "final_op"
type: "string"
allowed_values {
list {
s: "Id"
s: "Div"
}
}
}
attr {
name: "communication_hint"
type: "string"
default_value {
s: "auto"
}
}
attr {
name: "timeout_seconds"
type: "float"
default_value {
f: 0
}
}
attr {
name: "Nordering_token"
type: "int"
default_value {
i: 0
}
has_minimum: true
}
is_stateful: true
}

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

@ -7502,6 +7502,11 @@ op {
name: "input"
type_attr: "T"
}
input_arg {
name: "ordering_token"
type: DT_RESOURCE
number_attr: "Nordering_token"
}
output_arg {
name: "data"
type_attr: "T"
@ -7549,6 +7554,14 @@ op {
f: 0
}
}
attr {
name: "Nordering_token"
type: "int"
default_value {
i: 0
}
has_minimum: true
}
is_stateful: true
}
op {
@ -7569,6 +7582,11 @@ op {
name: "instance_key"
type: DT_INT32
}
input_arg {
name: "ordering_token"
type: DT_RESOURCE
number_attr: "Nordering_token"
}
output_arg {
name: "data"
type_attr: "T"
@ -7600,6 +7618,14 @@ op {
f: 0
}
}
attr {
name: "Nordering_token"
type: "int"
default_value {
i: 0
}
has_minimum: true
}
is_stateful: true
}
op {
@ -7648,6 +7674,11 @@ op {
name: "input"
type_attr: "T"
}
input_arg {
name: "ordering_token"
type: DT_RESOURCE
number_attr: "Nordering_token"
}
output_arg {
name: "data"
type_attr: "T"
@ -7725,6 +7756,14 @@ op {
f: 0
}
}
attr {
name: "Nordering_token"
type: "int"
default_value {
i: 0
}
has_minimum: true
}
is_stateful: true
}
op {
@ -7745,6 +7784,11 @@ op {
name: "instance_key"
type: DT_INT32
}
input_arg {
name: "ordering_token"
type: DT_RESOURCE
number_attr: "Nordering_token"
}
output_arg {
name: "data"
type_attr: "T"
@ -7798,6 +7842,14 @@ op {
f: 0
}
}
attr {
name: "Nordering_token"
type: "int"
default_value {
i: 0
}
has_minimum: true
}
is_stateful: true
}
op {