Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/tensorflow/core/protobuf/tpu/tpu_embedding_configuration_pb2.py: 42%

1# -*- coding: utf-8 -*- 

2# Generated by the protocol buffer compiler. DO NOT EDIT! 

3# source: tensorflow/core/protobuf/tpu/tpu_embedding_configuration.proto 

4"""Generated protocol buffer code.""" 

5from google.protobuf.internal import builder as _builder 

6from google.protobuf import descriptor as _descriptor 

7from google.protobuf import descriptor_pool as _descriptor_pool 

8from google.protobuf import symbol_database as _symbol_database 

9# @@protoc_insertion_point(imports) 

10 

11_sym_db = _symbol_database.Default() 

12 

13 

14from tensorflow.core.protobuf.tpu import optimization_parameters_pb2 as tensorflow_dot_core_dot_protobuf_dot_tpu_dot_optimization__parameters__pb2 

15 

16 

17DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n>tensorflow/core/protobuf/tpu/tpu_embedding_configuration.proto\x12\x0etensorflow.tpu\x1a:tensorflow/core/protobuf/tpu/optimization_parameters.proto\"\x93\x08\n\x19TPUEmbeddingConfiguration\x12S\n\x10table_descriptor\x18\x01 \x03(\x0b\x32\x39.tensorflow.tpu.TPUEmbeddingConfiguration.TableDescriptor\x12<\n\x04mode\x18\x02 \x01(\x0e\x32..tensorflow.tpu.TPUEmbeddingConfiguration.Mode\x12\"\n\x1a\x62\x61tch_size_per_tensor_core\x18\x03 \x01(\x05\x12\x11\n\tnum_hosts\x18\x04 \x01(\x05\x12\x18\n\x10num_tensor_cores\x18\x05 \x01(\x05\x12U\n\x11sharding_strategy\x18\x06 \x01(\x0e\x32:.tensorflow.tpu.TPUEmbeddingConfiguration.ShardingStrategy\x12+\n#pipeline_execution_with_tensor_core\x18\x07 \x01(\x08\x12\x1e\n\x16profile_data_directory\x18\t \x01(\t\x12W\n\x12\x66\x65\x61ture_descriptor\x18\n \x03(\x0b\x32;.tensorflow.tpu.TPUEmbeddingConfiguration.FeatureDescriptor\x12M\n\rspmd_sharding\x18\x0b \x01(\x0b\x32\x36.tensorflow.tpu.TPUEmbeddingConfiguration.SpmdSharding\x1a\xaa\x01\n\x0fTableDescriptor\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x17\n\x0fvocabulary_size\x18\x02 \x01(\x03\x12\x11\n\tdimension\x18\x03 \x01(\x05\x12\x14\n\x0cnum_features\x18\x04 \x01(\x05\x12G\n\x17optimization_parameters\x18\x05 \x01(\x0b\x32&.tensorflow.tpu.OptimizationParameters\x1aH\n\x11\x46\x65\x61tureDescriptor\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x10\n\x08table_id\x18\x02 \x01(\x05\x12\x13\n\x0binput_shape\x18\x03 \x03(\x05\x1a>\n\x0cSpmdSharding\x12\x0f\n\x07\x65nabled\x18\x01 \x01(\x08\x12\x1d\n\x15num_cores_per_replica\x18\x02 \x01(\x05\"L\n\x04Mode\x12\x0f\n\x0bUNSPECIFIED\x10\x00\x12\r\n\tINFERENCE\x10\x01\x12\x0c\n\x08TRAINING\x10\x02\x12\x16\n\x12\x42\x41\x43KWARD_PASS_ONLY\x10\x03\",\n\x10ShardingStrategy\x12\x0f\n\x0b\x44IV_DEFAULT\x10\x00\x12\x07\n\x03MOD\x10\x01J\x04\x08\x08\x10\tR\routput_layout\"\x13\n\x11TPUEmbeddingErrorb\x06proto3') 

18 

19_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals()) 

20_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'tensorflow.core.protobuf.tpu.tpu_embedding_configuration_pb2', globals()) 

21if _descriptor._USE_C_DESCRIPTORS == False: 

22 

23 DESCRIPTOR._options = None 

24 _TPUEMBEDDINGCONFIGURATION._serialized_start=143 

25 _TPUEMBEDDINGCONFIGURATION._serialized_end=1186 

26 _TPUEMBEDDINGCONFIGURATION_TABLEDESCRIPTOR._serialized_start=733 

27 _TPUEMBEDDINGCONFIGURATION_TABLEDESCRIPTOR._serialized_end=903 

28 _TPUEMBEDDINGCONFIGURATION_FEATUREDESCRIPTOR._serialized_start=905 

29 _TPUEMBEDDINGCONFIGURATION_FEATUREDESCRIPTOR._serialized_end=977 

30 _TPUEMBEDDINGCONFIGURATION_SPMDSHARDING._serialized_start=979 

31 _TPUEMBEDDINGCONFIGURATION_SPMDSHARDING._serialized_end=1041 

32 _TPUEMBEDDINGCONFIGURATION_MODE._serialized_start=1043 

33 _TPUEMBEDDINGCONFIGURATION_MODE._serialized_end=1119 

34 _TPUEMBEDDINGCONFIGURATION_SHARDINGSTRATEGY._serialized_start=1121 

35 _TPUEMBEDDINGCONFIGURATION_SHARDINGSTRATEGY._serialized_end=1165 

36 _TPUEMBEDDINGERROR._serialized_start=1188 

37 _TPUEMBEDDINGERROR._serialized_end=1207 

38# @@protoc_insertion_point(module_scope)