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