Update ops-related pbtxt files.
PiperOrigin-RevId: 234218960
This commit is contained in:
parent
745028b1fb
commit
a3399e17fd
@ -79928,6 +79928,118 @@ op {
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "TPUReplicate"
|
||||
input_arg {
|
||||
name: "inputs"
|
||||
type_list_attr: "Tinputs"
|
||||
}
|
||||
input_arg {
|
||||
name: "broadcast_inputs"
|
||||
type_list_attr: "Tbroadcast_inputs"
|
||||
}
|
||||
input_arg {
|
||||
name: "variables"
|
||||
type: DT_RESOURCE
|
||||
number_attr: "NumVariables"
|
||||
}
|
||||
input_arg {
|
||||
name: "guaranteed_constants"
|
||||
type_list_attr: "Tguaranteed_constants"
|
||||
}
|
||||
output_arg {
|
||||
name: "outputs"
|
||||
type_list_attr: "output_types"
|
||||
}
|
||||
attr {
|
||||
name: "computation"
|
||||
type: "func"
|
||||
}
|
||||
attr {
|
||||
name: "num_replicas"
|
||||
type: "int"
|
||||
has_minimum: true
|
||||
minimum: 1
|
||||
}
|
||||
attr {
|
||||
name: "num_cores_per_replica"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 1
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "topology"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: ""
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "use_tpu"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: true
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "device_assignment"
|
||||
type: "list(int)"
|
||||
default_value {
|
||||
list {
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "host_compute_core"
|
||||
type: "list(string)"
|
||||
default_value {
|
||||
list {
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "Tinputs"
|
||||
type: "list(type)"
|
||||
has_minimum: true
|
||||
}
|
||||
attr {
|
||||
name: "Tbroadcast_inputs"
|
||||
type: "list(type)"
|
||||
has_minimum: true
|
||||
}
|
||||
attr {
|
||||
name: "NumVariables"
|
||||
type: "int"
|
||||
has_minimum: true
|
||||
}
|
||||
attr {
|
||||
name: "Tguaranteed_constants"
|
||||
type: "list(type)"
|
||||
has_minimum: true
|
||||
}
|
||||
attr {
|
||||
name: "output_types"
|
||||
type: "list(type)"
|
||||
has_minimum: true
|
||||
}
|
||||
attr {
|
||||
name: "padding_map"
|
||||
type: "list(string)"
|
||||
default_value {
|
||||
list {
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "step_marker_location"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "STEP_MARK_AT_ENTRY"
|
||||
}
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "TPUReplicateMetadata"
|
||||
attr {
|
||||
@ -79989,6 +80101,74 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "TPUReplicateMetadata"
|
||||
attr {
|
||||
name: "num_replicas"
|
||||
type: "int"
|
||||
has_minimum: true
|
||||
}
|
||||
attr {
|
||||
name: "num_cores_per_replica"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 1
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "topology"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: ""
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "use_tpu"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: true
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "device_assignment"
|
||||
type: "list(int)"
|
||||
default_value {
|
||||
list {
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "computation_shape"
|
||||
type: "list(int)"
|
||||
default_value {
|
||||
list {
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "host_compute_core"
|
||||
type: "list(string)"
|
||||
default_value {
|
||||
list {
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "padding_map"
|
||||
type: "list(string)"
|
||||
default_value {
|
||||
list {
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "step_marker_location"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "STEP_MARK_AT_ENTRY"
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "TPUReplicatedInput"
|
||||
input_arg {
|
||||
|
@ -39409,6 +39409,13 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "step_marker_location"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "STEP_MARK_AT_ENTRY"
|
||||
}
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
@ -39471,6 +39478,13 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "step_marker_location"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "STEP_MARK_AT_ENTRY"
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "TPUReplicatedInput"
|
||||
|
Loading…
Reference in New Issue
Block a user