diff options
Diffstat (limited to 'tensorflow/core/protobuf/master.proto')
-rw-r--r-- | tensorflow/core/protobuf/master.proto | 68 |
1 files changed, 68 insertions, 0 deletions
diff --git a/tensorflow/core/protobuf/master.proto b/tensorflow/core/protobuf/master.proto index 0437cb1b83..96c91536f7 100644 --- a/tensorflow/core/protobuf/master.proto +++ b/tensorflow/core/protobuf/master.proto @@ -23,6 +23,7 @@ option java_package = "org.tensorflow.distruntime"; import "tensorflow/core/framework/device_attributes.proto"; import "tensorflow/core/framework/graph.proto"; +import "tensorflow/core/framework/tensor.proto"; import "tensorflow/core/lib/core/error_codes.proto"; import "tensorflow/core/protobuf/config.proto"; import "tensorflow/core/protobuf/named_tensor.proto"; @@ -264,3 +265,70 @@ message ListDevicesResponse { repeated DeviceAttributes local_device = 1; repeated DeviceAttributes remote_device = 2; } + +//////////////////////////////////////////////////////////////////////////////// +// +// MakeCallable method request/response protos. +// +//////////////////////////////////////////////////////////////////////////////// + +message MakeCallableRequest { + // REQUIRED: session_handle must be returned by a CreateSession call + // to the same master service. + string session_handle = 1; + + // Options that define the behavior of the created callable. + CallableOptions options = 2; +} + +message MakeCallableResponse { + // A handle to the created callable. + int64 handle = 1; +} + +//////////////////////////////////////////////////////////////////////////////// +// +// RunCallable method request/response protos. +// +//////////////////////////////////////////////////////////////////////////////// + +message RunCallableRequest { + // REQUIRED: session_handle must be returned by a CreateSession call + // to the same master service. + string session_handle = 1; + // REQUIRED: handle must be returned by a MakeCallable call to the same + // master service. + int64 handle = 2; + + // Values of the tensors passed as arguments to the callable, in the order + // defined in the CallableOptions.feed field passed to MakeCallable. + repeated TensorProto feed = 3; +} + +message RunCallableResponse { + // Values of the tensors returned by the callable, in the order defined in the + // CallableOptions.fetch field passed to MakeCallable. + repeated TensorProto fetch = 1; + + // Returned metadata if requested in the options. + RunMetadata metadata = 2; +} + +//////////////////////////////////////////////////////////////////////////////// +// +// ReleaseCallable method request/response protos. +// +//////////////////////////////////////////////////////////////////////////////// + +message ReleaseCallableRequest { + // REQUIRED: session_handle must be returned by a CreateSession call + // to the same master service. + string session_handle = 1; + + // REQUIRED: handle must be returned by a MakeCallable call to the same + // master service. + int64 handle = 2; +} + +message ReleaseCallableResponse { +} |