easy-cs-rec-custommodel 0.8.6__py2.py3-none-any.whl

easy_rec/python/protos/keras_layer_pb2.py

@@ -0,0 +1,692 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: easy_rec/python/protos/keras_layer.proto
+
+import sys
+
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pb2
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
+from google.protobuf import symbol_database as _symbol_database
+
+from easy_rec.python.protos import dnn_pb2 as easy__rec_dot_python_dot_protos_dot_dnn__pb2  # NOQA
+from easy_rec.python.protos import fm_pb2 as easy__rec_dot_python_dot_protos_dot_fm__pb2  # NOQA
+from easy_rec.python.protos import layer_pb2 as easy__rec_dot_python_dot_protos_dot_layer__pb2  # NOQA
+from easy_rec.python.protos import seq_encoder_pb2 as easy__rec_dot_python_dot_protos_dot_seq__encoder__pb2  # NOQA
+
+_b = sys.version_info[0] < 3 and (lambda x: x) or (lambda x: x.encode('latin1'))
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+DESCRIPTOR = _descriptor.FileDescriptor(
+    name='easy_rec/python/protos/keras_layer.proto',
+    package='protos',
+    syntax='proto2',
+    serialized_pb=_b(
+        '\n(easy_rec/python/protos/keras_layer.proto\x12\x06protos\x1a\x1cgoogle/protobuf/struct.proto\x1a\"easy_rec/python/protos/layer.proto\x1a easy_rec/python/protos/dnn.proto\x1a\x1f\x65\x61sy_rec/python/protos/fm.proto\x1a(easy_rec/python/protos/seq_encoder.proto\"\xef\x08\n\nKerasLayer\x12\x12\n\nclass_name\x18\x01 \x02(\t\x12,\n\tst_params\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructH\x00\x12\x37\n\x12periodic_embedding\x18\x03 \x01(\x0b\x32\x19.protos.PeriodicEmbeddingH\x00\x12\x36\n\x12\x61uto_dis_embedding\x18\x04 \x01(\x0b\x32\x18.protos.AutoDisEmbeddingH\x00\x12\x36\n\x12nary_dis_embedding\x18\x15 \x01(\x0b\x32\x18.protos.NaryDisEmbeddingH\x00\x12\x18\n\x02\x66m\x18\x05 \x01(\x0b\x32\n.protos.FMH\x00\x12\'\n\nmask_block\x18\x06 \x01(\x0b\x32\x11.protos.MaskBlockH\x00\x12\"\n\x07masknet\x18\x07 \x01(\x0b\x32\x0f.protos.MaskNetH\x00\x12\x1e\n\x05senet\x18\x08 \x01(\x0b\x32\r.protos.SENetH\x00\x12$\n\x08\x62ilinear\x18\t \x01(\x0b\x32\x10.protos.BilinearH\x00\x12\"\n\x07\x66ibinet\x18\n \x01(\x0b\x32\x0f.protos.FiBiNetH\x00\x12\x1a\n\x03mlp\x18\x0b \x01(\x0b\x32\x0b.protos.MLPH\x00\x12!\n\x03\x64in\x18\x0c \x01(\x0b\x32\x12.protos.DINEncoderH\x00\x12!\n\x03\x62st\x18\r \x01(\x0b\x32\x12.protos.BSTEncoderH\x00\x12!\n\x04mmoe\x18\x0e \x01(\x0b\x32\x11.protos.MMoELayerH\x00\x12*\n\x07seq_aug\x18\x0f \x01(\x0b\x32\x17.protos.SequenceAugmentH\x00\x12\x1e\n\x05ppnet\x18\x10 \x01(\x0b\x32\r.protos.PPNetH\x00\x12#\n\x08text_cnn\x18\x11 \x01(\x0b\x32\x0f.protos.TextCNNH\x00\x12\'\n\x07highway\x18\x12 \x01(\x0b\x32\x14.protos.HighWayTowerH\x00\x12)\n\x07overlap\x18\x13 \x01(\x0b\x32\x16.protos.OverlapFeatureH\x00\x12/\n\x0b\x64ot_product\x18\x14 \x01(\x0b\x32\x18.protos.MappedDotProductH\x00\x12&\n\tattention\x18\x16 \x01(\x0b\x32\x11.protos.AttentionH\x00\x12:\n\x14multi_head_attention\x18\x17 \x01(\x0b\x32\x1a.protos.MultiHeadAttentionH\x00\x12*\n\x0btransformer\x18\x18 \x01(\x0b\x32\x13.protos.TransformerH\x00\x12+\n\x0ctext_encoder\x18\x19 \x01(\x0b\x32\x13.protos.TextEncoderH\x00\x12$\n\x04gate\x18\x1a \x01(\x0b\x32\x14.protos.WeightedGateH\x00\x12!\n\x04\x61itm\x18\x1b \x01(\x0b\x32\x11.protos.AITMTowerH\x00\x12\x1a\n\x03\x63in\x18\x1c \x01(\x0b\x32\x0b.protos.CINH\x00\x42\x08\n\x06params'
+    ),
+    dependencies=[
+        google_dot_protobuf_dot_struct__pb2.DESCRIPTOR,
+        easy__rec_dot_python_dot_protos_dot_layer__pb2.DESCRIPTOR,
+        easy__rec_dot_python_dot_protos_dot_dnn__pb2.DESCRIPTOR,
+        easy__rec_dot_python_dot_protos_dot_fm__pb2.DESCRIPTOR,
+        easy__rec_dot_python_dot_protos_dot_seq__encoder__pb2.DESCRIPTOR,
+    ])
+
+_KERASLAYER = _descriptor.Descriptor(
+    name='KerasLayer',
+    full_name='protos.KerasLayer',
+    filename=None,
+    file=DESCRIPTOR,
+    containing_type=None,
+    fields=[
+        _descriptor.FieldDescriptor(
+            name='class_name',
+            full_name='protos.KerasLayer.class_name',
+            index=0,
+            number=1,
+            type=9,
+            cpp_type=9,
+            label=2,
+            has_default_value=False,
+            default_value=_b('').decode('utf-8'),
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='st_params',
+            full_name='protos.KerasLayer.st_params',
+            index=1,
+            number=2,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='periodic_embedding',
+            full_name='protos.KerasLayer.periodic_embedding',
+            index=2,
+            number=3,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='auto_dis_embedding',
+            full_name='protos.KerasLayer.auto_dis_embedding',
+            index=3,
+            number=4,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='nary_dis_embedding',
+            full_name='protos.KerasLayer.nary_dis_embedding',
+            index=4,
+            number=21,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='fm',
+            full_name='protos.KerasLayer.fm',
+            index=5,
+            number=5,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='mask_block',
+            full_name='protos.KerasLayer.mask_block',
+            index=6,
+            number=6,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='masknet',
+            full_name='protos.KerasLayer.masknet',
+            index=7,
+            number=7,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='senet',
+            full_name='protos.KerasLayer.senet',
+            index=8,
+            number=8,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='bilinear',
+            full_name='protos.KerasLayer.bilinear',
+            index=9,
+            number=9,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='fibinet',
+            full_name='protos.KerasLayer.fibinet',
+            index=10,
+            number=10,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='mlp',
+            full_name='protos.KerasLayer.mlp',
+            index=11,
+            number=11,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='din',
+            full_name='protos.KerasLayer.din',
+            index=12,
+            number=12,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='bst',
+            full_name='protos.KerasLayer.bst',
+            index=13,
+            number=13,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='mmoe',
+            full_name='protos.KerasLayer.mmoe',
+            index=14,
+            number=14,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='seq_aug',
+            full_name='protos.KerasLayer.seq_aug',
+            index=15,
+            number=15,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='ppnet',
+            full_name='protos.KerasLayer.ppnet',
+            index=16,
+            number=16,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='text_cnn',
+            full_name='protos.KerasLayer.text_cnn',
+            index=17,
+            number=17,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='highway',
+            full_name='protos.KerasLayer.highway',
+            index=18,
+            number=18,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='overlap',
+            full_name='protos.KerasLayer.overlap',
+            index=19,
+            number=19,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='dot_product',
+            full_name='protos.KerasLayer.dot_product',
+            index=20,
+            number=20,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='attention',
+            full_name='protos.KerasLayer.attention',
+            index=21,
+            number=22,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='multi_head_attention',
+            full_name='protos.KerasLayer.multi_head_attention',
+            index=22,
+            number=23,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='transformer',
+            full_name='protos.KerasLayer.transformer',
+            index=23,
+            number=24,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='text_encoder',
+            full_name='protos.KerasLayer.text_encoder',
+            index=24,
+            number=25,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='gate',
+            full_name='protos.KerasLayer.gate',
+            index=25,
+            number=26,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='aitm',
+            full_name='protos.KerasLayer.aitm',
+            index=26,
+            number=27,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='cin',
+            full_name='protos.KerasLayer.cin',
+            index=27,
+            number=28,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+    ],
+    extensions=[],
+    nested_types=[],
+    enum_types=[],
+    options=None,
+    is_extendable=False,
+    syntax='proto2',
+    extension_ranges=[],
+    oneofs=[
+        _descriptor.OneofDescriptor(
+            name='params',
+            full_name='protos.KerasLayer.params',
+            index=0,
+            containing_type=None,
+            fields=[]),
+    ],
+    serialized_start=228,
+    serialized_end=1363,
+)
+
+_KERASLAYER.fields_by_name[
+    'st_params'].message_type = google_dot_protobuf_dot_struct__pb2._STRUCT
+_KERASLAYER.fields_by_name[
+    'periodic_embedding'].message_type = easy__rec_dot_python_dot_protos_dot_layer__pb2._PERIODICEMBEDDING
+_KERASLAYER.fields_by_name[
+    'auto_dis_embedding'].message_type = easy__rec_dot_python_dot_protos_dot_layer__pb2._AUTODISEMBEDDING
+_KERASLAYER.fields_by_name[
+    'nary_dis_embedding'].message_type = easy__rec_dot_python_dot_protos_dot_layer__pb2._NARYDISEMBEDDING
+_KERASLAYER.fields_by_name[
+    'fm'].message_type = easy__rec_dot_python_dot_protos_dot_fm__pb2._FM
+_KERASLAYER.fields_by_name[
+    'mask_block'].message_type = easy__rec_dot_python_dot_protos_dot_layer__pb2._MASKBLOCK
+_KERASLAYER.fields_by_name[
+    'masknet'].message_type = easy__rec_dot_python_dot_protos_dot_layer__pb2._MASKNET
+_KERASLAYER.fields_by_name[
+    'senet'].message_type = easy__rec_dot_python_dot_protos_dot_layer__pb2._SENET
+_KERASLAYER.fields_by_name[
+    'bilinear'].message_type = easy__rec_dot_python_dot_protos_dot_layer__pb2._BILINEAR
+_KERASLAYER.fields_by_name[
+    'fibinet'].message_type = easy__rec_dot_python_dot_protos_dot_layer__pb2._FIBINET
+_KERASLAYER.fields_by_name[
+    'mlp'].message_type = easy__rec_dot_python_dot_protos_dot_dnn__pb2._MLP
+_KERASLAYER.fields_by_name[
+    'din'].message_type = easy__rec_dot_python_dot_protos_dot_seq__encoder__pb2._DINENCODER
+_KERASLAYER.fields_by_name[
+    'bst'].message_type = easy__rec_dot_python_dot_protos_dot_seq__encoder__pb2._BSTENCODER
+_KERASLAYER.fields_by_name[
+    'mmoe'].message_type = easy__rec_dot_python_dot_protos_dot_layer__pb2._MMOELAYER
+_KERASLAYER.fields_by_name[
+    'seq_aug'].message_type = easy__rec_dot_python_dot_protos_dot_seq__encoder__pb2._SEQUENCEAUGMENT
+_KERASLAYER.fields_by_name[
+    'ppnet'].message_type = easy__rec_dot_python_dot_protos_dot_layer__pb2._PPNET
+_KERASLAYER.fields_by_name[
+    'text_cnn'].message_type = easy__rec_dot_python_dot_protos_dot_layer__pb2._TEXTCNN
+_KERASLAYER.fields_by_name[
+    'highway'].message_type = easy__rec_dot_python_dot_protos_dot_layer__pb2._HIGHWAYTOWER
+_KERASLAYER.fields_by_name[
+    'overlap'].message_type = easy__rec_dot_python_dot_protos_dot_layer__pb2._OVERLAPFEATURE
+_KERASLAYER.fields_by_name[
+    'dot_product'].message_type = easy__rec_dot_python_dot_protos_dot_layer__pb2._MAPPEDDOTPRODUCT
+_KERASLAYER.fields_by_name[
+    'attention'].message_type = easy__rec_dot_python_dot_protos_dot_seq__encoder__pb2._ATTENTION
+_KERASLAYER.fields_by_name[
+    'multi_head_attention'].message_type = easy__rec_dot_python_dot_protos_dot_seq__encoder__pb2._MULTIHEADATTENTION
+_KERASLAYER.fields_by_name[
+    'transformer'].message_type = easy__rec_dot_python_dot_protos_dot_seq__encoder__pb2._TRANSFORMER
+_KERASLAYER.fields_by_name[
+    'text_encoder'].message_type = easy__rec_dot_python_dot_protos_dot_seq__encoder__pb2._TEXTENCODER
+_KERASLAYER.fields_by_name[
+    'gate'].message_type = easy__rec_dot_python_dot_protos_dot_layer__pb2._WEIGHTEDGATE
+_KERASLAYER.fields_by_name[
+    'aitm'].message_type = easy__rec_dot_python_dot_protos_dot_layer__pb2._AITMTOWER
+_KERASLAYER.fields_by_name[
+    'cin'].message_type = easy__rec_dot_python_dot_protos_dot_layer__pb2._CIN
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['st_params'])
+_KERASLAYER.fields_by_name[
+    'st_params'].containing_oneof = _KERASLAYER.oneofs_by_name['params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['periodic_embedding'])
+_KERASLAYER.fields_by_name[
+    'periodic_embedding'].containing_oneof = _KERASLAYER.oneofs_by_name[
+        'params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['auto_dis_embedding'])
+_KERASLAYER.fields_by_name[
+    'auto_dis_embedding'].containing_oneof = _KERASLAYER.oneofs_by_name[
+        'params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['nary_dis_embedding'])
+_KERASLAYER.fields_by_name[
+    'nary_dis_embedding'].containing_oneof = _KERASLAYER.oneofs_by_name[
+        'params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['fm'])
+_KERASLAYER.fields_by_name['fm'].containing_oneof = _KERASLAYER.oneofs_by_name[
+    'params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['mask_block'])
+_KERASLAYER.fields_by_name[
+    'mask_block'].containing_oneof = _KERASLAYER.oneofs_by_name['params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['masknet'])
+_KERASLAYER.fields_by_name[
+    'masknet'].containing_oneof = _KERASLAYER.oneofs_by_name['params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['senet'])
+_KERASLAYER.fields_by_name[
+    'senet'].containing_oneof = _KERASLAYER.oneofs_by_name['params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['bilinear'])
+_KERASLAYER.fields_by_name[
+    'bilinear'].containing_oneof = _KERASLAYER.oneofs_by_name['params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['fibinet'])
+_KERASLAYER.fields_by_name[
+    'fibinet'].containing_oneof = _KERASLAYER.oneofs_by_name['params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['mlp'])
+_KERASLAYER.fields_by_name['mlp'].containing_oneof = _KERASLAYER.oneofs_by_name[
+    'params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['din'])
+_KERASLAYER.fields_by_name['din'].containing_oneof = _KERASLAYER.oneofs_by_name[
+    'params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['bst'])
+_KERASLAYER.fields_by_name['bst'].containing_oneof = _KERASLAYER.oneofs_by_name[
+    'params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['mmoe'])
+_KERASLAYER.fields_by_name[
+    'mmoe'].containing_oneof = _KERASLAYER.oneofs_by_name['params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['seq_aug'])
+_KERASLAYER.fields_by_name[
+    'seq_aug'].containing_oneof = _KERASLAYER.oneofs_by_name['params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['ppnet'])
+_KERASLAYER.fields_by_name[
+    'ppnet'].containing_oneof = _KERASLAYER.oneofs_by_name['params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['text_cnn'])
+_KERASLAYER.fields_by_name[
+    'text_cnn'].containing_oneof = _KERASLAYER.oneofs_by_name['params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['highway'])
+_KERASLAYER.fields_by_name[
+    'highway'].containing_oneof = _KERASLAYER.oneofs_by_name['params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['overlap'])
+_KERASLAYER.fields_by_name[
+    'overlap'].containing_oneof = _KERASLAYER.oneofs_by_name['params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['dot_product'])
+_KERASLAYER.fields_by_name[
+    'dot_product'].containing_oneof = _KERASLAYER.oneofs_by_name['params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['attention'])
+_KERASLAYER.fields_by_name[
+    'attention'].containing_oneof = _KERASLAYER.oneofs_by_name['params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['multi_head_attention'])
+_KERASLAYER.fields_by_name[
+    'multi_head_attention'].containing_oneof = _KERASLAYER.oneofs_by_name[
+        'params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['transformer'])
+_KERASLAYER.fields_by_name[
+    'transformer'].containing_oneof = _KERASLAYER.oneofs_by_name['params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['text_encoder'])
+_KERASLAYER.fields_by_name[
+    'text_encoder'].containing_oneof = _KERASLAYER.oneofs_by_name['params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['gate'])
+_KERASLAYER.fields_by_name[
+    'gate'].containing_oneof = _KERASLAYER.oneofs_by_name['params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['aitm'])
+_KERASLAYER.fields_by_name[
+    'aitm'].containing_oneof = _KERASLAYER.oneofs_by_name['params']
+_KERASLAYER.oneofs_by_name['params'].fields.append(
+    _KERASLAYER.fields_by_name['cin'])
+_KERASLAYER.fields_by_name['cin'].containing_oneof = _KERASLAYER.oneofs_by_name[
+    'params']
+DESCRIPTOR.message_types_by_name['KerasLayer'] = _KERASLAYER
+_sym_db.RegisterFileDescriptor(DESCRIPTOR)
+
+KerasLayer = _reflection.GeneratedProtocolMessageType(
+    'KerasLayer',
+    (_message.Message,),
+    dict(
+        DESCRIPTOR=_KERASLAYER,
+        __module__='easy_rec.python.protos.keras_layer_pb2'
+        # @@protoc_insertion_point(class_scope:protos.KerasLayer)
+    ))
+_sym_db.RegisterMessage(KerasLayer)
+
+# @@protoc_insertion_point(module_scope)