Update ops-related pbtxt files.
PiperOrigin-RevId: 400325502
Change-Id: Ic89b619d4a43c98fcdba64707c48d3e1d848be7e
diff --git a/tensorflow/core/ops/compat/ops_history_v2/TPUExecute.pbtxt b/tensorflow/core/ops/compat/ops_history_v2/TPUExecute.pbtxt
index f60da0a..97a2c2a 100644
--- a/tensorflow/core/ops/compat/ops_history_v2/TPUExecute.pbtxt
+++ b/tensorflow/core/ops/compat/ops_history_v2/TPUExecute.pbtxt
@@ -24,3 +24,30 @@
}
is_stateful: true
}
+op {
+ name: "TPUExecute"
+ input_arg {
+ name: "args"
+ type_list_attr: "Targs"
+ }
+ input_arg {
+ name: "key"
+ type: DT_STRING
+ }
+ output_arg {
+ name: "results"
+ type_list_attr: "Tresults"
+ }
+ attr {
+ name: "Targs"
+ type: "list(type)"
+ has_minimum: true
+ }
+ attr {
+ name: "Tresults"
+ type: "list(type)"
+ has_minimum: true
+ }
+ is_stateful: true
+ is_distributed_communication: true
+}
diff --git a/tensorflow/core/ops/compat/ops_history_v2/TPUExecuteAndUpdateVariables.pbtxt b/tensorflow/core/ops/compat/ops_history_v2/TPUExecuteAndUpdateVariables.pbtxt
index 8fab665..5a611f9 100644
--- a/tensorflow/core/ops/compat/ops_history_v2/TPUExecuteAndUpdateVariables.pbtxt
+++ b/tensorflow/core/ops/compat/ops_history_v2/TPUExecuteAndUpdateVariables.pbtxt
@@ -34,3 +34,40 @@
}
is_stateful: true
}
+op {
+ name: "TPUExecuteAndUpdateVariables"
+ input_arg {
+ name: "args"
+ type_list_attr: "Targs"
+ }
+ input_arg {
+ name: "key"
+ type: DT_STRING
+ }
+ output_arg {
+ name: "results"
+ type_list_attr: "Tresults"
+ }
+ attr {
+ name: "Targs"
+ type: "list(type)"
+ has_minimum: true
+ }
+ attr {
+ name: "Tresults"
+ type: "list(type)"
+ has_minimum: true
+ }
+ attr {
+ name: "device_var_reads_indices"
+ type: "list(int)"
+ has_minimum: true
+ }
+ attr {
+ name: "device_var_updates_indices"
+ type: "list(int)"
+ has_minimum: true
+ }
+ is_stateful: true
+ is_distributed_communication: true
+}
diff --git a/tensorflow/core/ops/ops.pbtxt b/tensorflow/core/ops/ops.pbtxt
index a44c00f..cf3436f 100644
--- a/tensorflow/core/ops/ops.pbtxt
+++ b/tensorflow/core/ops/ops.pbtxt
@@ -54835,6 +54835,7 @@
has_minimum: true
}
is_stateful: true
+ is_distributed_communication: true
}
op {
name: "TPUExecuteAndUpdateVariables"
@@ -54871,6 +54872,7 @@
has_minimum: true
}
is_stateful: true
+ is_distributed_communication: true
}
op {
name: "TPUOrdinalSelector"
