tensorflow.graph_transfer_info source code

1 // Generated by the protocol buffer compiler.  DO NOT EDIT!
2 // source: tensorflow/core/framework/graph_transfer_info.proto
3 
4 module tensorflow.graph_transfer_info;
5 
6 import google.protobuf;
7 import tensorflow.types;
8 
9 enum protocVersion = 3012004;
10 
11 class GraphTransferNodeInput
12 {
13     @Proto(1) int nodeId = protoDefaultValue!(int);
14     @Proto(2) int outputPort = protoDefaultValue!(int);
15 }
16 
17 class GraphTransferNodeInfo
18 {
19     @Proto(1) string name = protoDefaultValue!(string);
20     @Proto(2) int nodeId = protoDefaultValue!(int);
21     @Proto(3) string typeName = protoDefaultValue!(string);
22     @Proto(4) int socOpId = protoDefaultValue!(int);
23     @Proto(5) int paddingId = protoDefaultValue!(int);
24     @Proto(6) int inputCount = protoDefaultValue!(int);
25     @Proto(7) int outputCount = protoDefaultValue!(int);
26 }
27 
28 class GraphTransferConstNodeInfo
29 {
30     @Proto(1) string name = protoDefaultValue!(string);
31     @Proto(2) int nodeId = protoDefaultValue!(int);
32     @Proto(3, Wire.none, Yes.packed) long[] shape = protoDefaultValue!(long[]);
33     @Proto(4) bytes data = protoDefaultValue!(bytes);
34     @Proto(5) DataType dtype = protoDefaultValue!(DataType);
35 }
36 
37 class GraphTransferNodeInputInfo
38 {
39     @Proto(1) int nodeId = protoDefaultValue!(int);
40     @Proto(2) GraphTransferNodeInput[] nodeInput = protoDefaultValue!(GraphTransferNodeInput[]);
41 }
42 
43 class GraphTransferNodeOutputInfo
44 {
45     @Proto(1) int nodeId = protoDefaultValue!(int);
46     @Proto(2, Wire.none, Yes.packed) int[] maxByteSize = protoDefaultValue!(int[]);
47 }
48 
49 class GraphTransferGraphInputNodeInfo
50 {
51     @Proto(1) string name = protoDefaultValue!(string);
52     @Proto(2, Wire.none, Yes.packed) long[] shape = protoDefaultValue!(long[]);
53     @Proto(3) DataType dtype = protoDefaultValue!(DataType);
54 }
55 
56 class GraphTransferGraphOutputNodeInfo
57 {
58     @Proto(1) string name = protoDefaultValue!(string);
59     @Proto(2, Wire.none, Yes.packed) long[] shape = protoDefaultValue!(long[]);
60     @Proto(3) DataType dtype = protoDefaultValue!(DataType);
61 }
62 
63 class GraphTransferInfo
64 {
65     @Proto(1) GraphTransferNodeInfo[] nodeInfo = protoDefaultValue!(GraphTransferNodeInfo[]);
66     @Proto(2) GraphTransferConstNodeInfo[] constNodeInfo = protoDefaultValue!(GraphTransferConstNodeInfo[]);
67     @Proto(3) GraphTransferNodeInputInfo[] nodeInputInfo = protoDefaultValue!(GraphTransferNodeInputInfo[]);
68     @Proto(4) GraphTransferNodeOutputInfo[] nodeOutputInfo = protoDefaultValue!(GraphTransferNodeOutputInfo[]);
69     @Proto(5) GraphTransferGraphInputNodeInfo[] graphInputNodeInfo = protoDefaultValue!(GraphTransferGraphInputNodeInfo[]);
70     @Proto(6) GraphTransferGraphOutputNodeInfo[] graphOutputNodeInfo = protoDefaultValue!(GraphTransferGraphOutputNodeInfo[]);
71     @Proto(7) Destination destination = protoDefaultValue!(Destination);
72 
73     enum Destination
74     {
75         NOP = 0,
76         HEXAGON = 1,
77     }
78 }