tensorflow.meta_graph source code

1 // Generated by the protocol buffer compiler.  DO NOT EDIT!
2 // source: tensorflow/core/protobuf/meta_graph.proto
3 
4 module tensorflow.meta_graph;
5 
6 import google.protobuf;
7 import google.protobuf.any;
8 import tensorflow.graph;
9 import tensorflow.op_def;
10 import tensorflow.tensor_shape;
11 import tensorflow.types;
12 import tensorflow.saved_object_graph;
13 import tensorflow.saver;
14 import tensorflow.struct_;
15 
16 enum protocVersion = 3012004;
17 
18 class MetaGraphDef
19 {
20     @Proto(1) MetaGraphDef.MetaInfoDef metaInfoDef = protoDefaultValue!(MetaGraphDef.MetaInfoDef);
21     @Proto(2) GraphDef graphDef = protoDefaultValue!(GraphDef);
22     @Proto(3) SaverDef saverDef = protoDefaultValue!(SaverDef);
23     @Proto(4) CollectionDef[string] collectionDef = protoDefaultValue!(CollectionDef[string]);
24     @Proto(5) SignatureDef[string] signatureDef = protoDefaultValue!(SignatureDef[string]);
25     @Proto(6) AssetFileDef[] assetFileDef = protoDefaultValue!(AssetFileDef[]);
26     @Proto(7) SavedObjectGraph objectGraphDef = protoDefaultValue!(SavedObjectGraph);
27 
28     static class MetaInfoDef
29     {
30         @Proto(1) string metaGraphVersion = protoDefaultValue!(string);
31         @Proto(2) OpList strippedOpList = protoDefaultValue!(OpList);
32         @Proto(3) Any anyInfo = protoDefaultValue!(Any);
33         @Proto(4) string[] tags = protoDefaultValue!(string[]);
34         @Proto(5) string tensorflowVersion = protoDefaultValue!(string);
35         @Proto(6) string tensorflowGitVersion = protoDefaultValue!(string);
36         @Proto(7) bool strippedDefaultAttrs = protoDefaultValue!(bool);
37     }
38 }
39 
40 class CollectionDef
41 {
42     enum KindCase
43     {
44         kindNotSet = 0,
45         nodeList = 1,
46         bytesList = 2,
47         int64List = 3,
48         floatList = 4,
49         anyList = 5,
50     }
51     KindCase _kindCase = KindCase.kindNotSet;
52     @property KindCase kindCase() { return _kindCase; }
53     void clearKind() { _kindCase = KindCase.kindNotSet; }
54     @Oneof("_kindCase") union
55     {
56         @Proto(1) CollectionDef.NodeList _nodeList = protoDefaultValue!(CollectionDef.NodeList); mixin(oneofAccessors!_nodeList);
57         @Proto(2) CollectionDef.BytesList _bytesList; mixin(oneofAccessors!_bytesList);
58         @Proto(3) CollectionDef.Int64List _int64List; mixin(oneofAccessors!_int64List);
59         @Proto(4) CollectionDef.FloatList _floatList; mixin(oneofAccessors!_floatList);
60         @Proto(5) CollectionDef.AnyList _anyList; mixin(oneofAccessors!_anyList);
61     }
62 
63     static class NodeList
64     {
65         @Proto(1) string[] value = protoDefaultValue!(string[]);
66     }
67 
68     static class BytesList
69     {
70         @Proto(1) bytes[] value = protoDefaultValue!(bytes[]);
71     }
72 
73     static class Int64List
74     {
75         @Proto(1, Wire.none, Yes.packed) long[] value = protoDefaultValue!(long[]);
76     }
77 
78     static class FloatList
79     {
80         @Proto(1, Wire.none, Yes.packed) float[] value = protoDefaultValue!(float[]);
81     }
82 
83     static class AnyList
84     {
85         @Proto(1) Any[] value = protoDefaultValue!(Any[]);
86     }
87 }
88 
89 class TensorInfo
90 {
91     enum EncodingCase
92     {
93         encodingNotSet = 0,
94         name = 1,
95         cooSparse = 4,
96         compositeTensor = 5,
97     }
98     EncodingCase _encodingCase = EncodingCase.encodingNotSet;
99     @property EncodingCase encodingCase() { return _encodingCase; }
100     void clearEncoding() { _encodingCase = EncodingCase.encodingNotSet; }
101     @Oneof("_encodingCase") union
102     {
103         @Proto(1) string _name = protoDefaultValue!(string); mixin(oneofAccessors!_name);
104         @Proto(4) TensorInfo.CooSparse _cooSparse; mixin(oneofAccessors!_cooSparse);
105         @Proto(5) TensorInfo.CompositeTensor _compositeTensor; mixin(oneofAccessors!_compositeTensor);
106     }
107     @Proto(2) DataType dtype = protoDefaultValue!(DataType);
108     @Proto(3) TensorShapeProto tensorShape = protoDefaultValue!(TensorShapeProto);
109 
110     static class CooSparse
111     {
112         @Proto(1) string valuesTensorName = protoDefaultValue!(string);
113         @Proto(2) string indicesTensorName = protoDefaultValue!(string);
114         @Proto(3) string denseShapeTensorName = protoDefaultValue!(string);
115     }
116 
117     static class CompositeTensor
118     {
119         @Proto(1) TypeSpecProto typeSpec = protoDefaultValue!(TypeSpecProto);
120         @Proto(2) TensorInfo[] components = protoDefaultValue!(TensorInfo[]);
121     }
122 }
123 
124 class SignatureDef
125 {
126     @Proto(1) TensorInfo[string] inputs = protoDefaultValue!(TensorInfo[string]);
127     @Proto(2) TensorInfo[string] outputs = protoDefaultValue!(TensorInfo[string]);
128     @Proto(3) string methodName = protoDefaultValue!(string);
129 }
130 
131 class AssetFileDef
132 {
133     @Proto(1) TensorInfo tensorInfo = protoDefaultValue!(TensorInfo);
134     @Proto(2) string filename = protoDefaultValue!(string);
135 }