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
|
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 {
|
op {
|
||||||
name: "TPUReplicateMetadata"
|
name: "TPUReplicateMetadata"
|
||||||
attr {
|
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 {
|
op {
|
||||||
name: "TPUReplicatedInput"
|
name: "TPUReplicatedInput"
|
||||||
input_arg {
|
input_arg {
|
||||||
|
@ -39409,6 +39409,13 @@ op {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
attr {
|
||||||
|
name: "step_marker_location"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: "STEP_MARK_AT_ENTRY"
|
||||||
|
}
|
||||||
|
}
|
||||||
is_stateful: true
|
is_stateful: true
|
||||||
}
|
}
|
||||||
op {
|
op {
|
||||||
@ -39471,6 +39478,13 @@ op {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
attr {
|
||||||
|
name: "step_marker_location"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: "STEP_MARK_AT_ENTRY"
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
op {
|
op {
|
||||||
name: "TPUReplicatedInput"
|
name: "TPUReplicatedInput"
|
||||||
|
Loading…
Reference in New Issue
Block a user