Update ops-related pbtxt files.
PiperOrigin-RevId: 192155883
This commit is contained in:
parent
800f1db6f0
commit
568f05c793
@ -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 {
|
||||
|
@ -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 {
|
||||
|
Loading…
Reference in New Issue
Block a user