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