Update ops-related pbtxt files.
PiperOrigin-RevId: 319887474 Change-Id: I03c2bd78d98491ab55a8d17deb0a9bf572f91015
This commit is contained in:
parent
91417e256e
commit
331ce4d203
@ -202,3 +202,85 @@ 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"
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "allow_soft_placement"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: false
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "use_spmd_for_xla_partitioning"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: false
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@ -50770,6 +50770,13 @@ op {
|
||||
b: false
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "use_spmd_for_xla_partitioning"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: false
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "TPUReplicatedInput"
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user