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 {
name: "input"
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 {
name: "input"