tensorflow.eager.eager_service source code

1 // Generated by the protocol buffer compiler.  DO NOT EDIT!
2 // source: tensorflow/core/protobuf/eager_service.proto
3 
4 module tensorflow.eager.eager_service;
5 
6 import google.protobuf;
7 import tensorflow.attr_value;
8 import tensorflow.device_attributes;
9 import tensorflow.function_;
10 import tensorflow.tensor;
11 import tensorflow.tensor_shape;
12 import tensorflow.versions;
13 import tensorflow.eager.remote_tensor_handle;
14 import tensorflow.tensorflow_server;
15 
16 enum protocVersion = 3012004;
17 
18 class Operation
19 {
20     @Proto(1) long id = protoDefaultValue!(long);
21     @Proto(2) string name = protoDefaultValue!(string);
22     @Proto(3) RemoteTensorHandle[] inputs = protoDefaultValue!(RemoteTensorHandle[]);
23     @Proto(4, Wire.none, Yes.packed) long[] controlOpIds = protoDefaultValue!(long[]);
24     @Proto(5) AttrValue[string] attrs = protoDefaultValue!(AttrValue[string]);
25     @Proto(6) string device = protoDefaultValue!(string);
26     @Proto(7) bool isComponentFunction = protoDefaultValue!(bool);
27     @Proto(8) long funcStepId = protoDefaultValue!(long);
28     @Proto(9) bool isFunction = protoDefaultValue!(bool);
29 }
30 
31 class QueueItem
32 {
33     enum ItemCase
34     {
35         itemNotSet = 0,
36         handleToDecref = 1,
37         operation = 2,
38         sendTensor = 3,
39         registerFunction = 4,
40         cleanupFunction = 5,
41     }
42     ItemCase _itemCase = ItemCase.itemNotSet;
43     @property ItemCase itemCase() { return _itemCase; }
44     void clearItem() { _itemCase = ItemCase.itemNotSet; }
45     @Oneof("_itemCase") union
46     {
47         @Proto(1) RemoteTensorHandle _handleToDecref = protoDefaultValue!(RemoteTensorHandle); mixin(oneofAccessors!_handleToDecref);
48         @Proto(2) Operation _operation; mixin(oneofAccessors!_operation);
49         @Proto(3) SendTensorOp _sendTensor; mixin(oneofAccessors!_sendTensor);
50         @Proto(4) RegisterFunctionOp _registerFunction; mixin(oneofAccessors!_registerFunction);
51         @Proto(5) CleanupFunctionOp _cleanupFunction; mixin(oneofAccessors!_cleanupFunction);
52     }
53 }
54 
55 class QueueResponse
56 {
57     @Proto(1) TensorShapeProto[] shape = protoDefaultValue!(TensorShapeProto[]);
58 }
59 
60 class CreateContextRequest
61 {
62     @Proto(1) ServerDef serverDef = protoDefaultValue!(ServerDef);
63     @Proto(2) bool async = protoDefaultValue!(bool);
64     @Proto(3) long keepAliveSecs = protoDefaultValue!(long);
65     @Proto(4) VersionDef versionDef = protoDefaultValue!(VersionDef);
66     @Proto(6) DeviceAttributes[] clusterDeviceAttributes = protoDefaultValue!(DeviceAttributes[]);
67     @Proto(7, Wire.fixed) ulong contextId = protoDefaultValue!(ulong);
68     @Proto(8, Wire.fixed) ulong contextViewId = protoDefaultValue!(ulong);
69     @Proto(9) bool lazyCopyRemoteFunctionInputs = protoDefaultValue!(bool);
70 }
71 
72 class CreateContextResponse
73 {
74     @Proto(2) DeviceAttributes[] deviceAttributes = protoDefaultValue!(DeviceAttributes[]);
75 }
76 
77 class UpdateContextRequest
78 {
79     @Proto(1) ServerDef serverDef = protoDefaultValue!(ServerDef);
80     @Proto(2) DeviceAttributes[] clusterDeviceAttributes = protoDefaultValue!(DeviceAttributes[]);
81     @Proto(3, Wire.fixed) ulong contextId = protoDefaultValue!(ulong);
82     @Proto(4, Wire.fixed) ulong contextViewId = protoDefaultValue!(ulong);
83 }
84 
85 class UpdateContextResponse
86 {
87     @Proto(1) DeviceAttributes[] deviceAttributes = protoDefaultValue!(DeviceAttributes[]);
88 }
89 
90 class EnqueueRequest
91 {
92     @Proto(1, Wire.fixed) ulong contextId = protoDefaultValue!(ulong);
93     @Proto(3) QueueItem[] queue = protoDefaultValue!(QueueItem[]);
94 }
95 
96 class EnqueueResponse
97 {
98     @Proto(1) QueueResponse[] queueResponse = protoDefaultValue!(QueueResponse[]);
99 }
100 
101 class WaitQueueDoneRequest
102 {
103     @Proto(1, Wire.fixed) ulong contextId = protoDefaultValue!(ulong);
104     @Proto(2, Wire.none, Yes.packed) long[] opId = protoDefaultValue!(long[]);
105 }
106 
107 class WaitQueueDoneResponse
108 {
109 }
110 
111 class KeepAliveRequest
112 {
113     @Proto(1, Wire.fixed) ulong contextId = protoDefaultValue!(ulong);
114 }
115 
116 class KeepAliveResponse
117 {
118     @Proto(1, Wire.fixed) ulong contextViewId = protoDefaultValue!(ulong);
119 }
120 
121 class CloseContextRequest
122 {
123     @Proto(1, Wire.fixed) ulong contextId = protoDefaultValue!(ulong);
124     @Proto(2, Wire.fixed) ulong contextViewId = protoDefaultValue!(ulong);
125 }
126 
127 class CloseContextResponse
128 {
129 }
130 
131 class RegisterFunctionOp
132 {
133     @Proto(1) FunctionDef functionDef = protoDefaultValue!(FunctionDef);
134     @Proto(2) bool isComponentFunction = protoDefaultValue!(bool);
135     @Proto(3) FunctionDefLibrary library = protoDefaultValue!(FunctionDefLibrary);
136 }
137 
138 class CleanupFunctionOp
139 {
140     @Proto(1) long stepId = protoDefaultValue!(long);
141 }
142 
143 class SendTensorOp
144 {
145     @Proto(1) long opId = protoDefaultValue!(long);
146     @Proto(2) TensorProto[] tensors = protoDefaultValue!(TensorProto[]);
147     @Proto(3) string deviceName = protoDefaultValue!(string);
148 }