diff options
-rw-r--r-- | tensorflow/core/ops/compat/ops_history.v1.pbtxt | 31 | ||||
-rw-r--r-- | tensorflow/core/ops/ops.pbtxt | 38 |
2 files changed, 69 insertions, 0 deletions
diff --git a/tensorflow/core/ops/compat/ops_history.v1.pbtxt b/tensorflow/core/ops/compat/ops_history.v1.pbtxt index e76573ffdb..6ff1a3fc03 100644 --- a/tensorflow/core/ops/compat/ops_history.v1.pbtxt +++ b/tensorflow/core/ops/compat/ops_history.v1.pbtxt @@ -19997,6 +19997,37 @@ op { } } op { + name: "RemoteCall" + input_arg { + name: "target" + type: DT_STRING + } + input_arg { + name: "args" + type_list_attr: "Tin" + } + output_arg { + name: "output" + type_list_attr: "Tout" + } + attr { + name: "Tin" + type: "list(type)" + has_minimum: true + minimum: 1 + } + attr { + name: "Tout" + type: "list(type)" + has_minimum: true + minimum: 1 + } + attr { + name: "f" + type: "func" + } +} +op { name: "RemoteFusedGraphExecute" input_arg { name: "inputs" diff --git a/tensorflow/core/ops/ops.pbtxt b/tensorflow/core/ops/ops.pbtxt index 06eabdcdcd..87cdc30fb1 100644 --- a/tensorflow/core/ops/ops.pbtxt +++ b/tensorflow/core/ops/ops.pbtxt @@ -19608,6 +19608,44 @@ op { summary: "Computes rectified linear gradients for a Relu operation." } op { + name: "RemoteCall" + input_arg { + name: "target" + description: "A fully specified device name where we want to run the function." + type: DT_STRING + } + input_arg { + name: "args" + description: "A list of arguments for the function." + type_list_attr: "Tin" + } + output_arg { + name: "output" + description: "A list of return values." + type_list_attr: "Tout" + } + attr { + name: "Tin" + type: "list(type)" + description: "The type list for the arguments." + has_minimum: true + minimum: 1 + } + attr { + name: "Tout" + type: "list(type)" + description: "The type list for the return values." + has_minimum: true + minimum: 1 + } + attr { + name: "f" + type: "func" + description: "The function to run remotely." + } + summary: "Runs function `f` on a remote device indicated by `target`." +} +op { name: "RemoteFusedGraphExecute" input_arg { name: "inputs" |