1 syntax = "proto3"; 2 3 package tensorflow; 4 option cc_enable_arenas = true; 5 option java_outer_classname = "RemoteFusedGraphExecuteInfoProto"; 6 option java_multiple_files = true; 7 option java_package = "org.tensorflow.framework"; 8 9 import "tensorflow/core/framework/graph.proto"; 10 import "tensorflow/core/framework/tensor_shape.proto"; 11 import "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. 16 message RemoteFusedGraphExecuteInfo { 17 enum NodeType { 18 UNUSED = 0; 19 GRAPH_INPUT = 1; 20 GRAPH_OUTPUT = 2; 21 FUSED_NODE = 3; 22 BORDER_INPUT = 4; 23 BORDER_OUTPUT = 5; 24 } 25 26 message TensorShapeTypeProto { 27 DataType dtype = 1; 28 TensorShapeProto shape = 2; 29 } 30 31 // Definition of remote graph 32 GraphDef remote_graph = 1; 33 34 // Remote fused graph input node name 35 repeated string graph_input_node_name = 2; 36 37 // Remote fused graph output node name 38 repeated string graph_output_node_name = 3; 39 40 // Executor's name 41 string executor_name = 4; 42 43 // Optional: Parameters given to the executor 44 bytes serialized_executor_parameters = 5; 45 46 // Optional: Default graph input tensor shape used to allocate memory 47 // before executing op 48 repeated TensorShapeTypeProto default_graph_input_tensor_shape = 6; 49 50 // Optional: Default graph input tensor shape used to allocate memory 51 // before executing op 52 // TODO(satok): Remote output tensor shape once shape information is stored 53 // in NodeDef 54 repeated TensorShapeTypeProto default_graph_output_tensor_shape = 7; 55 }; 56
