• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1syntax = "proto3";
2
3package tensorflow;
4option cc_enable_arenas = true;
5option java_outer_classname = "RemoteFusedGraphExecuteInfoProto";
6option java_multiple_files = true;
7option java_package = "org.tensorflow.framework";
8option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
9import "tensorflow/core/framework/graph.proto";
10import "tensorflow/core/framework/tensor_shape.proto";
11import "tensorflow/core/framework/types.proto";
12
13// Protocol buffer representing a handle to a tensorflow resource. Handles are
14// not valid across executions, but can be serialized back and forth from within
15// a single run.
16message RemoteFusedGraphExecuteInfo {
17
18  message TensorShapeTypeProto {
19    DataType dtype = 1;
20    TensorShapeProto shape = 2;
21  }
22
23  // Definition of remote graph
24  GraphDef remote_graph = 1;
25
26  // Remote fused graph input node name
27  repeated string graph_input_node_name = 2;
28
29  // Remote fused graph output node name
30  repeated string graph_output_node_name = 3;
31
32  // Executor's name
33  string executor_name = 4;
34
35  // Optional: Parameters given to the executor
36  bytes serialized_executor_parameters = 5;
37
38  // Optional: Default graph input tensor shape used to allocate memory
39  // before executing op
40  repeated TensorShapeTypeProto default_graph_input_tensor_shape = 6;
41
42  // Optional: Default graph input tensor shape used to allocate memory
43  // before executing op
44  // TODO(satok): Remote output tensor shape once shape information is stored
45  // in NodeDef
46  repeated TensorShapeTypeProto default_graph_output_tensor_shape = 7;
47};
48