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

Update ops-related pbtxt files.

Browse Source 
PiperOrigin-RevId: 348070251
Change-Id: I13c09a925289acb6839e0d8f91c5a34945da0734
This commit is contained in:
A. Unique TensorFlower 2020-12-17 12:18:54 -08:00 committed by TensorFlower Gardener
parent a4c338bd76
commit 48d3d6eb47
3 changed files with 234 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

65
tensorflow/core/ops/compat/ops_history_v2/CollectiveBcastRecvV2.pbtxt Normal file
View File

@ -0,0 +1,65 @@
op {
name: "CollectiveBcastRecvV2"
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: "shape"
type_attr: "Tshape"
}
output_arg {
name: "data"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BOOL
type: DT_FLOAT
type: DT_HALF
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "Tshape"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
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
}
}
is_stateful: true
}

52
tensorflow/core/ops/compat/ops_history_v2/CollectiveBcastSendV2.pbtxt Normal file
View File

@ -0,0 +1,52 @@
op {
name: "CollectiveBcastSendV2"
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
}
output_arg {
name: "data"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BOOL
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
}
}
is_stateful: true
}

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

@ -7441,6 +7441,71 @@ op {
}
is_stateful: true
}
op {
name: "CollectiveBcastRecvV2"
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: "shape"
type_attr: "Tshape"
}
output_arg {
name: "data"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BOOL
type: DT_FLOAT
type: DT_HALF
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "Tshape"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
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
}
}
is_stateful: true
}
op {
name: "CollectiveBcastSend"
input_arg {
@ -7497,6 +7562,58 @@ op {
}
is_stateful: true
}
op {
name: "CollectiveBcastSendV2"
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
}
output_arg {
name: "data"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BOOL
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
}
}
is_stateful: true
}
op {
name: "CollectiveGather"
input_arg {