google-research

1
# coding=utf-8
2
# Copyright 2024 The Google Research Authors.
3
#
4
# Licensed under the Apache License, Version 2.0 (the "License");
5
# you may not use this file except in compliance with the License.
6
# You may obtain a copy of the License at
7
#
8
#     http://www.apache.org/licenses/LICENSE-2.0
9
#
10
# Unless required by applicable law or agreed to in writing, software
11
# distributed under the License is distributed on an "AS IS" BASIS,
12
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
# See the License for the specific language governing permissions and
14
# limitations under the License.
15

16
# pylint: skip-file
17
# Generated by the protocol buffer compiler.  DO NOT EDIT!
18
# source: inception_cell_spec.proto
19

20
import sys
21
_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
22
from google.protobuf import descriptor as _descriptor
23
from google.protobuf import message as _message
24
from google.protobuf import reflection as _reflection
25
from google.protobuf import symbol_database as _symbol_database
26
from google.protobuf import descriptor_pb2
27
# @@protoc_insertion_point(imports)
28

29
_sym_db = _symbol_database.Default()
30

31

32

33

34
DESCRIPTOR = _descriptor.FileDescriptor(
35
  name='inception_cell_spec.proto',
36
  package='google_research.evanet',
37
  syntax='proto2',
38
  serialized_pb=_b('\n\x19inception_cell_spec.proto\x12\x16google_research.evanet\"\xb6\x01\n\x07\x44NASpec\x12\r\n\x05time1\x18\x01 \x02(\x05\x12\x10\n\x08\x64ilation\x18\x02 \x02(\x05\x12\x0f\n\x07\x63onvop1\x18\x03 \x02(\x05\x12\x16\n\x0emax_pool1_time\x18\x04 \x02(\x05\x12\x16\n\x0emax_pool3_time\x18\x05 \x02(\x05\x12\x16\n\x0emax_pool4_time\x18\x06 \x02(\x05\x12\x31\n\x06\x62locks\x18\x07 \x03(\x0b\x32!.google_research.evanet.BlockSpec\"[\n\tBlockSpec\x12\x0f\n\x07repeats\x18\x01 \x02(\x05\x12\x0e\n\x06use_nl\x18\x02 \x02(\x08\x12-\n\x06layers\x18\x03 \x03(\x0b\x32\x1d.google_research.evanet.Layer\"\xfa\x02\n\x05Layer\x12;\n\nlayer_type\x18\x01 \x02(\x0e\x32\'.google_research.evanet.Layer.LayerType\x12\x39\n\tconv_type\x18\x02 \x01(\x0e\x32&.google_research.evanet.Layer.ConvType\x12\x0c\n\x04time\x18\x03 \x01(\x05\x12\x10\n\x08\x64ilation\x18\x04 \x01(\x05\x12\r\n\x05time2\x18\x05 \x01(\x05\x12\x11\n\tdilation2\x18\x06 \x01(\x05\x12:\n\nconv_type2\x18\x07 \x01(\x0e\x32&.google_research.evanet.Layer.ConvType\"E\n\tLayerType\x12\x12\n\x0eSINGLE_DEFAULT\x10\x00\x12\x08\n\x04\x43ONV\x10\x01\x12\t\n\x05\x43ONV2\x10\x02\x12\x0f\n\x0bMAXPOOLCONV\x10\x03\"4\n\x08\x43onvType\x12\x12\n\x0e\x43ONV3D_DEFAULT\x10\x00\x12\x07\n\x03TGM\x10\x01\x12\x0b\n\x07\x43ONV2P1\x10\x02')
39
)
40
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
41

42

43

44
_LAYER_LAYERTYPE = _descriptor.EnumDescriptor(
45
  name='LayerType',
46
  full_name='google_research.evanet.Layer.LayerType',
47
  filename=None,
48
  file=DESCRIPTOR,
49
  values=[
50
    _descriptor.EnumValueDescriptor(
51
      name='SINGLE_DEFAULT', index=0, number=0,
52
      options=None,
53
      type=None),
54
    _descriptor.EnumValueDescriptor(
55
      name='CONV', index=1, number=1,
56
      options=None,
57
      type=None),
58
    _descriptor.EnumValueDescriptor(
59
      name='CONV2', index=2, number=2,
60
      options=None,
61
      type=None),
62
    _descriptor.EnumValueDescriptor(
63
      name='MAXPOOLCONV', index=3, number=3,
64
      options=None,
65
      type=None),
66
  ],
67
  containing_type=None,
68
  options=None,
69
  serialized_start=587,
70
  serialized_end=656,
71
)
72
_sym_db.RegisterEnumDescriptor(_LAYER_LAYERTYPE)
73

74
_LAYER_CONVTYPE = _descriptor.EnumDescriptor(
75
  name='ConvType',
76
  full_name='google_research.evanet.Layer.ConvType',
77
  filename=None,
78
  file=DESCRIPTOR,
79
  values=[
80
    _descriptor.EnumValueDescriptor(
81
      name='CONV3D_DEFAULT', index=0, number=0,
82
      options=None,
83
      type=None),
84
    _descriptor.EnumValueDescriptor(
85
      name='TGM', index=1, number=1,
86
      options=None,
87
      type=None),
88
    _descriptor.EnumValueDescriptor(
89
      name='CONV2P1', index=2, number=2,
90
      options=None,
91
      type=None),
92
  ],
93
  containing_type=None,
94
  options=None,
95
  serialized_start=658,
96
  serialized_end=710,
97
)
98
_sym_db.RegisterEnumDescriptor(_LAYER_CONVTYPE)
99

100

101
_DNASPEC = _descriptor.Descriptor(
102
  name='DNASpec',
103
  full_name='google_research.evanet.DNASpec',
104
  filename=None,
105
  file=DESCRIPTOR,
106
  containing_type=None,
107
  fields=[
108
    _descriptor.FieldDescriptor(
109
      name='time1', full_name='google_research.evanet.DNASpec.time1', index=0,
110
      number=1, type=5, cpp_type=1, label=2,
111
      has_default_value=False, default_value=0,
112
      message_type=None, enum_type=None, containing_type=None,
113
      is_extension=False, extension_scope=None,
114
      options=None),
115
    _descriptor.FieldDescriptor(
116
      name='dilation', full_name='google_research.evanet.DNASpec.dilation', index=1,
117
      number=2, type=5, cpp_type=1, label=2,
118
      has_default_value=False, default_value=0,
119
      message_type=None, enum_type=None, containing_type=None,
120
      is_extension=False, extension_scope=None,
121
      options=None),
122
    _descriptor.FieldDescriptor(
123
      name='convop1', full_name='google_research.evanet.DNASpec.convop1', index=2,
124
      number=3, type=5, cpp_type=1, label=2,
125
      has_default_value=False, default_value=0,
126
      message_type=None, enum_type=None, containing_type=None,
127
      is_extension=False, extension_scope=None,
128
      options=None),
129
    _descriptor.FieldDescriptor(
130
      name='max_pool1_time', full_name='google_research.evanet.DNASpec.max_pool1_time', index=3,
131
      number=4, type=5, cpp_type=1, label=2,
132
      has_default_value=False, default_value=0,
133
      message_type=None, enum_type=None, containing_type=None,
134
      is_extension=False, extension_scope=None,
135
      options=None),
136
    _descriptor.FieldDescriptor(
137
      name='max_pool3_time', full_name='google_research.evanet.DNASpec.max_pool3_time', index=4,
138
      number=5, type=5, cpp_type=1, label=2,
139
      has_default_value=False, default_value=0,
140
      message_type=None, enum_type=None, containing_type=None,
141
      is_extension=False, extension_scope=None,
142
      options=None),
143
    _descriptor.FieldDescriptor(
144
      name='max_pool4_time', full_name='google_research.evanet.DNASpec.max_pool4_time', index=5,
145
      number=6, type=5, cpp_type=1, label=2,
146
      has_default_value=False, default_value=0,
147
      message_type=None, enum_type=None, containing_type=None,
148
      is_extension=False, extension_scope=None,
149
      options=None),
150
    _descriptor.FieldDescriptor(
151
      name='blocks', full_name='google_research.evanet.DNASpec.blocks', index=6,
152
      number=7, type=11, cpp_type=10, label=3,
153
      has_default_value=False, default_value=[],
154
      message_type=None, enum_type=None, containing_type=None,
155
      is_extension=False, extension_scope=None,
156
      options=None),
157
  ],
158
  extensions=[
159
  ],
160
  nested_types=[],
161
  enum_types=[
162
  ],
163
  options=None,
164
  is_extendable=False,
165
  syntax='proto2',
166
  extension_ranges=[],
167
  oneofs=[
168
  ],
169
  serialized_start=54,
170
  serialized_end=236,
171
)
172

173

174
_BLOCKSPEC = _descriptor.Descriptor(
175
  name='BlockSpec',
176
  full_name='google_research.evanet.BlockSpec',
177
  filename=None,
178
  file=DESCRIPTOR,
179
  containing_type=None,
180
  fields=[
181
    _descriptor.FieldDescriptor(
182
      name='repeats', full_name='google_research.evanet.BlockSpec.repeats', index=0,
183
      number=1, type=5, cpp_type=1, label=2,
184
      has_default_value=False, default_value=0,
185
      message_type=None, enum_type=None, containing_type=None,
186
      is_extension=False, extension_scope=None,
187
      options=None),
188
    _descriptor.FieldDescriptor(
189
      name='use_nl', full_name='google_research.evanet.BlockSpec.use_nl', index=1,
190
      number=2, type=8, cpp_type=7, label=2,
191
      has_default_value=False, default_value=False,
192
      message_type=None, enum_type=None, containing_type=None,
193
      is_extension=False, extension_scope=None,
194
      options=None),
195
    _descriptor.FieldDescriptor(
196
      name='layers', full_name='google_research.evanet.BlockSpec.layers', index=2,
197
      number=3, type=11, cpp_type=10, label=3,
198
      has_default_value=False, default_value=[],
199
      message_type=None, enum_type=None, containing_type=None,
200
      is_extension=False, extension_scope=None,
201
      options=None),
202
  ],
203
  extensions=[
204
  ],
205
  nested_types=[],
206
  enum_types=[
207
  ],
208
  options=None,
209
  is_extendable=False,
210
  syntax='proto2',
211
  extension_ranges=[],
212
  oneofs=[
213
  ],
214
  serialized_start=238,
215
  serialized_end=329,
216
)
217

218

219
_LAYER = _descriptor.Descriptor(
220
  name='Layer',
221
  full_name='google_research.evanet.Layer',
222
  filename=None,
223
  file=DESCRIPTOR,
224
  containing_type=None,
225
  fields=[
226
    _descriptor.FieldDescriptor(
227
      name='layer_type', full_name='google_research.evanet.Layer.layer_type', index=0,
228
      number=1, type=14, cpp_type=8, label=2,
229
      has_default_value=False, default_value=0,
230
      message_type=None, enum_type=None, containing_type=None,
231
      is_extension=False, extension_scope=None,
232
      options=None),
233
    _descriptor.FieldDescriptor(
234
      name='conv_type', full_name='google_research.evanet.Layer.conv_type', index=1,
235
      number=2, type=14, cpp_type=8, label=1,
236
      has_default_value=False, default_value=0,
237
      message_type=None, enum_type=None, containing_type=None,
238
      is_extension=False, extension_scope=None,
239
      options=None),
240
    _descriptor.FieldDescriptor(
241
      name='time', full_name='google_research.evanet.Layer.time', index=2,
242
      number=3, type=5, cpp_type=1, label=1,
243
      has_default_value=False, default_value=0,
244
      message_type=None, enum_type=None, containing_type=None,
245
      is_extension=False, extension_scope=None,
246
      options=None),
247
    _descriptor.FieldDescriptor(
248
      name='dilation', full_name='google_research.evanet.Layer.dilation', index=3,
249
      number=4, type=5, cpp_type=1, label=1,
250
      has_default_value=False, default_value=0,
251
      message_type=None, enum_type=None, containing_type=None,
252
      is_extension=False, extension_scope=None,
253
      options=None),
254
    _descriptor.FieldDescriptor(
255
      name='time2', full_name='google_research.evanet.Layer.time2', index=4,
256
      number=5, type=5, cpp_type=1, label=1,
257
      has_default_value=False, default_value=0,
258
      message_type=None, enum_type=None, containing_type=None,
259
      is_extension=False, extension_scope=None,
260
      options=None),
261
    _descriptor.FieldDescriptor(
262
      name='dilation2', full_name='google_research.evanet.Layer.dilation2', index=5,
263
      number=6, type=5, cpp_type=1, label=1,
264
      has_default_value=False, default_value=0,
265
      message_type=None, enum_type=None, containing_type=None,
266
      is_extension=False, extension_scope=None,
267
      options=None),
268
    _descriptor.FieldDescriptor(
269
      name='conv_type2', full_name='google_research.evanet.Layer.conv_type2', index=6,
270
      number=7, type=14, cpp_type=8, label=1,
271
      has_default_value=False, default_value=0,
272
      message_type=None, enum_type=None, containing_type=None,
273
      is_extension=False, extension_scope=None,
274
      options=None),
275
  ],
276
  extensions=[
277
  ],
278
  nested_types=[],
279
  enum_types=[
280
    _LAYER_LAYERTYPE,
281
    _LAYER_CONVTYPE,
282
  ],
283
  options=None,
284
  is_extendable=False,
285
  syntax='proto2',
286
  extension_ranges=[],
287
  oneofs=[
288
  ],
289
  serialized_start=332,
290
  serialized_end=710,
291
)
292

293
_DNASPEC.fields_by_name['blocks'].message_type = _BLOCKSPEC
294
_BLOCKSPEC.fields_by_name['layers'].message_type = _LAYER
295
_LAYER.fields_by_name['layer_type'].enum_type = _LAYER_LAYERTYPE
296
_LAYER.fields_by_name['conv_type'].enum_type = _LAYER_CONVTYPE
297
_LAYER.fields_by_name['conv_type2'].enum_type = _LAYER_CONVTYPE
298
_LAYER_LAYERTYPE.containing_type = _LAYER
299
_LAYER_CONVTYPE.containing_type = _LAYER
300
DESCRIPTOR.message_types_by_name['DNASpec'] = _DNASPEC
301
DESCRIPTOR.message_types_by_name['BlockSpec'] = _BLOCKSPEC
302
DESCRIPTOR.message_types_by_name['Layer'] = _LAYER
303

304
DNASpec = _reflection.GeneratedProtocolMessageType('DNASpec', (_message.Message,), dict(
305
  DESCRIPTOR = _DNASPEC,
306
  __module__ = 'inception_cell_spec_pb2'
307
  # @@protoc_insertion_point(class_scope:google_research.evanet.DNASpec)
308
  ))
309
_sym_db.RegisterMessage(DNASpec)
310

311
BlockSpec = _reflection.GeneratedProtocolMessageType('BlockSpec', (_message.Message,), dict(
312
  DESCRIPTOR = _BLOCKSPEC,
313
  __module__ = 'inception_cell_spec_pb2'
314
  # @@protoc_insertion_point(class_scope:google_research.evanet.BlockSpec)
315
  ))
316
_sym_db.RegisterMessage(BlockSpec)
317

318
Layer = _reflection.GeneratedProtocolMessageType('Layer', (_message.Message,), dict(
319
  DESCRIPTOR = _LAYER,
320
  __module__ = 'inception_cell_spec_pb2'
321
  # @@protoc_insertion_point(class_scope:google_research.evanet.Layer)
322
  ))
323
_sym_db.RegisterMessage(Layer)
324

325

326
# @@protoc_insertion_point(module_scope)
327