# -*- coding: utf-8 -*-
# Generated by the protocol buffer compiler.  DO NOT EDIT!
# source: jax_tpu_embedding/sparsecore/lib/proto/embedding_spec.proto
"""Generated protocol buffer code."""
from google.protobuf.internal import builder as _builder
from google.protobuf import descriptor as _descriptor
from google.protobuf import descriptor_pool as _descriptor_pool
from google.protobuf import symbol_database as _symbol_database
# @@protoc_insertion_point(imports)

_sym_db = _symbol_database.Default()




DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n;jax_tpu_embedding/sparsecore/lib/proto/embedding_spec.proto\x12\x35third_party.py.jax_tpu_embedding.sparsecore_lib_proto\"\x8e\x01\n\x10\x46\x65\x61tureSpecProto\x12\x14\n\x0c\x66\x65\x61ture_name\x18\x01 \x01(\t\x12\x13\n\x0binput_shape\x18\x02 \x03(\x03\x12\x14\n\x0coutput_shape\x18\x03 \x03(\x03\x12\x12\n\nrow_offset\x18\x04 \x01(\x03\x12\x12\n\ncol_offset\x18\x05 \x01(\x03\x12\x11\n\tcol_shift\x18\x06 \x01(\x03\"\xe2\x02\n\x0eTableSpecProto\x12\x12\n\ntable_name\x18\x01 \x01(\t\x12\x12\n\nvocab_size\x18\x02 \x01(\x03\x12\x15\n\rembedding_dim\x18\x03 \x01(\x03\x12\x1d\n\x15max_ids_per_partition\x18\x05 \x01(\x03\x12$\n\x1cmax_unique_ids_per_partition\x18\x06 \x01(\x03\x12\x19\n\x11padded_vocab_size\x18\x07 \x01(\x03\x12\x1c\n\x14padded_embedding_dim\x18\x08 \x01(\x03\x12\x1b\n\x13row_offset_in_shard\x18\t \x01(\x03\x12\x16\n\x0eshard_rotation\x18\n \x01(\x03\x12^\n\rfeature_specs\x18\x0b \x03(\x0b\x32G.third_party.py.jax_tpu_embedding.sparsecore_lib_proto.FeatureSpecProto\"\xb8\x02\n\x15StackedTableSpecProto\x12\x12\n\nstack_name\x18\x01 \x01(\t\x12\x18\n\x10stack_vocab_size\x18\x02 \x01(\x03\x12\x1b\n\x13stack_embedding_dim\x18\x03 \x01(\x03\x12\x1a\n\x12total_sample_count\x18\x06 \x01(\x03\x12\x1d\n\x15max_ids_per_partition\x18\x07 \x01(\x03\x12$\n\x1cmax_unique_ids_per_partition\x18\x08 \x01(\x03\x12\x17\n\x0fnum_sparsecores\x18\t \x01(\x03\x12Z\n\x0btable_specs\x18\n \x03(\x0b\x32\x45.third_party.py.jax_tpu_embedding.sparsecore_lib_proto.TableSpecProto\"\x7f\n\x12\x45mbeddingSpecProto\x12i\n\x13stacked_table_specs\x18\x01 \x03(\x0b\x32L.third_party.py.jax_tpu_embedding.sparsecore_lib_proto.StackedTableSpecProtoB\x02P\x01\x62\x06proto3')

_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'jax_tpu_embedding.sparsecore.lib.proto.embedding_spec_pb2', globals())
if _descriptor._USE_C_DESCRIPTORS == False:

  DESCRIPTOR._options = None
  DESCRIPTOR._serialized_options = b'P\001'
  _FEATURESPECPROTO._serialized_start=119
  _FEATURESPECPROTO._serialized_end=261
  _TABLESPECPROTO._serialized_start=264
  _TABLESPECPROTO._serialized_end=618
  _STACKEDTABLESPECPROTO._serialized_start=621
  _STACKEDTABLESPECPROTO._serialized_end=933
  _EMBEDDINGSPECPROTO._serialized_start=935
  _EMBEDDINGSPECPROTO._serialized_end=1062
# @@protoc_insertion_point(module_scope)
