Update ops-related pbtxt files.

PiperOrigin-RevId: 319887474
Change-Id: I03c2bd78d98491ab55a8d17deb0a9bf572f91015
This commit is contained in:
A. Unique TensorFlower 2020-07-06 17:18:38 -07:00 committed by TensorFlower Gardener
parent 91417e256e
commit 331ce4d203
2 changed files with 89 additions and 0 deletions

View File

@ -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
}
}
}

View File

@ -50770,6 +50770,13 @@ op {
b: false
}
}
attr {
name: "use_spmd_for_xla_partitioning"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "TPUReplicatedInput"