Update ops-related pbtxt files.

PiperOrigin-RevId: 234218960
This commit is contained in:
A. Unique TensorFlower 2019-02-15 14:58:48 -08:00 committed by TensorFlower Gardener
parent 745028b1fb
commit a3399e17fd
2 changed files with 194 additions and 0 deletions

View File

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

View File

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