Update ops-related pbtxt files.

PiperOrigin-RevId: 192155883
This commit is contained in:
A. Unique TensorFlower 2018-04-09 11:19:41 -07:00 committed by TensorFlower Gardener
parent 800f1db6f0
commit 568f05c793
2 changed files with 282 additions and 0 deletions

View File

@ -11645,6 +11645,147 @@ op {
}
is_stateful: true
}
op {
name: "CollectiveBcastRecv"
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"
}
is_stateful: true
}
op {
name: "CollectiveBcastSend"
input_arg {
name: "input"
type_attr: "T"
}
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"
}
is_stateful: true
}
op {
name: "CollectiveReduce"
input_arg {
name: "input"
type_attr: "T"
}
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)"
}
is_stateful: true
}
op {
name: "CompareAndBitpack"
input_arg {

View File

@ -4729,6 +4729,147 @@ op {
}
is_stateful: true
}
op {
name: "CollectiveBcastRecv"
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"
}
is_stateful: true
}
op {
name: "CollectiveBcastSend"
input_arg {
name: "input"
type_attr: "T"
}
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"
}
is_stateful: true
}
op {
name: "CollectiveReduce"
input_arg {
name: "input"
type_attr: "T"
}
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)"
}
is_stateful: true
}
op {
name: "CompareAndBitpack"
input_arg {