o ®Ÿ?eã ã@s@dZddlZddlmZddlmZedƒGdd„deƒƒZdS)zTPU hardware feature info.éN)Ú topology_pb2)Ú tf_exportz tpu.experimental.HardwareFeaturec@sBeZdZdZdd„ZGdd„dejƒZedd„ƒZ e dd „ƒZ d S) ÚHardwareFeaturez/class holds all the feature info about the TPU.cCs ||_dS)zŒStore TPU hardware feature info. Args: tpu_hardware_feature_proto: protobuf which describe the tpu hardware feature. N)Útpu_hardware_feature_proto)Úselfr©rúk/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/tpu/tpu_hardware_feature.pyÚ__init__s zHardwareFeature.__init__c@seZdZdZdZdZdZdS)z HardwareFeature.EmbeddingFeatureaôEmbedding feature flag strings. UNSUPPORTED: No embedding lookup accelerator available on the tpu. V1: Embedding lookup accelerator V1. The embedding lookup operation can only be placed at the beginning of computation. Only one instance of embedding lookup layer is allowed. V2: Embedding lookup accelerator V2. The embedding lookup operation can be placed anywhere of the computation. Multiple instances of embedding lookup layer is allowed. Ú UNSUPPORTEDÚV1ÚV2N)Ú__name__Ú __module__Ú __qualname__Ú__doc__r r r rrrrÚEmbeddingFeature"s  rcCs>tjjjtjjtjjjtjjtjjjtjji}| |tjj¡S)z3Convert the embedding feature proto to enum string.)rZTPUHardwareFeaturerr rr r Úget)ÚclsZembedding_feature_protoZ%embedding_feature_proto_to_string_maprrrÚ"_embedding_feature_proto_to_string2súÿz2HardwareFeature._embedding_feature_proto_to_stringcCst |jj¡S)zITPU embedding feature. Returns: An EmbeddingFeature enum. )rrrÚembedding_feature)rrrrr@sÿz!HardwareFeature.embedding_featureN) r rrrr ÚenumÚEnumrÚ classmethodrÚpropertyrrrrrrs   r)rrZtensorflow.core.protobuf.tpurZ tensorflow.python.util.tf_exportrÚobjectrrrrrÚs