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