1 syntax = "proto3"; 2 3 package tensorflow; 4 option cc_enable_arenas = true; 5 option java_outer_classname = "GraphTransferInfoProto"; 6 option java_multiple_files = true; 7 option java_package = "org.tensorflow.framework"; 8 9 import "tensorflow/core/framework/types.proto"; 10 11 // Protocol buffer representing a handle to a tensorflow resource. Handles are 12 // not valid across executions, but can be serialized back and forth from within 13 // a single run. 14 message GraphTransferInfo { 15 enum Destination { 16 NOP = 0; 17 HEXAGON = 1; 18 } 19 message NodeInput { 20 int32 node_id = 1; 21 int32 output_port = 2; 22 } 23 message NodeInfo { 24 string name = 1; 25 int32 node_id = 2; 26 string type_name = 3; 27 int32 soc_op_id = 4; 28 int32 padding_id = 5; 29 int32 input_count = 6; 30 int32 output_count = 7; 31 }; 32 message ConstNodeInfo { 33 string name = 1; 34 int32 node_id = 2; 35 repeated int64 shape = 3; 36 bytes data = 4; 37 DataType dtype = 5; 38 }; 39 message NodeInputInfo { 40 int32 node_id = 1; 41 repeated NodeInput node_input = 2; 42 }; 43 message NodeOutputInfo { 44 int32 node_id = 1; 45 repeated int32 max_byte_size = 2; 46 }; 47 message GraphInputNodeInfo { 48 string name = 1; 49 repeated int64 shape = 2; 50 DataType dtype = 3; 51 } 52 53 message GraphOutputNodeInfo { 54 string name = 1; 55 repeated int64 shape = 2; 56 DataType dtype = 3; 57 } 58 59 repeated NodeInfo node_info = 1; 60 repeated ConstNodeInfo const_node_info = 2; 61 repeated NodeInputInfo node_input_info = 3; 62 repeated NodeOutputInfo node_output_info = 4; 63 // Input Node parameters of transferred graph 64 repeated GraphInputNodeInfo graph_input_node_info = 5; 65 repeated GraphOutputNodeInfo graph_output_node_info = 6; 66 // Destination of graph transfer 67 Destination destination = 7; 68 }; 69
