tensorflow.saved_object_graph source code

1 // Generated by the protocol buffer compiler.  DO NOT EDIT!
2 // source: tensorflow/core/protobuf/saved_object_graph.proto
3 
4 module tensorflow.saved_object_graph;
5 
6 import google.protobuf;
7 import tensorflow.trackable_object_graph;
8 import tensorflow.struct_;
9 import tensorflow.tensor_shape;
10 import tensorflow.types;
11 import tensorflow.versions;
12 import tensorflow.variable;
13 
14 enum protocVersion = 3012004;
15 
16 class SavedObjectGraph
17 {
18     @Proto(1) SavedObject[] nodes = protoDefaultValue!(SavedObject[]);
19     @Proto(2) SavedConcreteFunction[string] concreteFunctions = protoDefaultValue!(SavedConcreteFunction[string]);
20 }
21 
22 class SavedObject
23 {
24     @Proto(1) TrackableObjectGraph.TrackableObject.ObjectReference[] children = protoDefaultValue!(TrackableObjectGraph.TrackableObject.ObjectReference[]);
25     @Proto(3) TrackableObjectGraph.TrackableObject.SlotVariableReference[] slotVariables = protoDefaultValue!(TrackableObjectGraph.TrackableObject.SlotVariableReference[]);
26     enum KindCase
27     {
28         kindNotSet = 0,
29         userObject = 4,
30         asset = 5,
31         function_ = 6,
32         variable = 7,
33         bareConcreteFunction = 8,
34         constant = 9,
35         resource = 10,
36     }
37     KindCase _kindCase = KindCase.kindNotSet;
38     @property KindCase kindCase() { return _kindCase; }
39     void clearKind() { _kindCase = KindCase.kindNotSet; }
40     @Oneof("_kindCase") union
41     {
42         @Proto(4) SavedUserObject _userObject = protoDefaultValue!(SavedUserObject); mixin(oneofAccessors!_userObject);
43         @Proto(5) SavedAsset _asset; mixin(oneofAccessors!_asset);
44         @Proto(6) SavedFunction _function_; mixin(oneofAccessors!_function_);
45         @Proto(7) SavedVariable _variable; mixin(oneofAccessors!_variable);
46         @Proto(8) SavedBareConcreteFunction _bareConcreteFunction; mixin(oneofAccessors!_bareConcreteFunction);
47         @Proto(9) SavedConstant _constant; mixin(oneofAccessors!_constant);
48         @Proto(10) SavedResource _resource; mixin(oneofAccessors!_resource);
49     }
50 }
51 
52 class SavedUserObject
53 {
54     @Proto(1) string identifier = protoDefaultValue!(string);
55     @Proto(2) VersionDef version_ = protoDefaultValue!(VersionDef);
56     @Proto(3) string metadata = protoDefaultValue!(string);
57 }
58 
59 class SavedAsset
60 {
61     @Proto(1) int assetFileDefIndex = protoDefaultValue!(int);
62 }
63 
64 class SavedFunction
65 {
66     @Proto(1) string[] concreteFunctions = protoDefaultValue!(string[]);
67     @Proto(2) FunctionSpec functionSpec = protoDefaultValue!(FunctionSpec);
68 }
69 
70 class SavedConcreteFunction
71 {
72     @Proto(2, Wire.none, Yes.packed) int[] boundInputs = protoDefaultValue!(int[]);
73     @Proto(3) StructuredValue canonicalizedInputSignature = protoDefaultValue!(StructuredValue);
74     @Proto(4) StructuredValue outputSignature = protoDefaultValue!(StructuredValue);
75 }
76 
77 class SavedBareConcreteFunction
78 {
79     @Proto(1) string concreteFunctionName = protoDefaultValue!(string);
80     @Proto(2) string[] argumentKeywords = protoDefaultValue!(string[]);
81     @Proto(3) long allowedPositionalArguments = protoDefaultValue!(long);
82 }
83 
84 class SavedConstant
85 {
86     @Proto(1) string operation = protoDefaultValue!(string);
87 }
88 
89 class SavedVariable
90 {
91     @Proto(1) DataType dtype = protoDefaultValue!(DataType);
92     @Proto(2) TensorShapeProto shape = protoDefaultValue!(TensorShapeProto);
93     @Proto(3) bool trainable = protoDefaultValue!(bool);
94     @Proto(4) VariableSynchronization synchronization = protoDefaultValue!(VariableSynchronization);
95     @Proto(5) VariableAggregation aggregation = protoDefaultValue!(VariableAggregation);
96     @Proto(6) string name = protoDefaultValue!(string);
97 }
98 
99 class FunctionSpec
100 {
101     @Proto(1) StructuredValue fullargspec = protoDefaultValue!(StructuredValue);
102     @Proto(2) bool isMethod = protoDefaultValue!(bool);
103     @Proto(5) StructuredValue inputSignature = protoDefaultValue!(StructuredValue);
104 }
105 
106 class SavedResource
107 {
108     @Proto(1) string device = protoDefaultValue!(string);
109 }