BVh TdZddlZddlmZddlmZedGddeZy)zTPU hardware feature info.N) topology_pb2) tf_exportz tpu.experimental.HardwareFeaturecpeZdZdZdZGddej ZedZ e dZ e dZ y) HardwareFeaturez/class holds all the feature info about the TPU.c||_y)zStore TPU hardware feature info. Args: tpu_hardware_feature_proto: protobuf which describe the tpu hardware feature. N)tpu_hardware_feature_proto)selfrs Z/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/tpu/tpu_hardware_feature.py__init__zHardwareFeature.__init__s 'AD#ceZdZdZdZdZdZy) HardwareFeature.EmbeddingFeatureaEmbedding 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. UNSUPPORTEDV1V2N)__name__ __module__ __qualname____doc__rrrr r EmbeddingFeaturer"s  K B Br rctjjjtjjtjjj tjj tjjj tjj i}|j|tjjS)z3Convert the embedding feature proto to enum string.)rTPUHardwareFeaturerrrrrget)clsembedding_feature_proto%embedding_feature_proto_to_string_maps r "_embedding_feature_proto_to_stringz2HardwareFeature._embedding_feature_proto_to_string2s ''88DD  , , 8 8''88;;  , , / /''88;;  , , / / -) 1 4 4!A!A!M!M OOr cTtj|jjS)zITPU embedding feature. Returns: An EmbeddingFeature enum. )rrrembedding_featurer s r r z!HardwareFeature.embedding_feature@s&  = = ''99 ;;r c.|jjS)zpNumber of embedding accelerator devices per chip. Returns: Number of embedding devices per chip. )rnum_embedding_devices_per_chipr!s r r#z.HardwareFeature.num_embedding_devices_per_chipJs  * * I IIr N) rrrrr enumEnumr classmethodrpropertyr r#rr r rrs[7A  O O ; ; J Jr r)rr$tensorflow.core.protobuf.tpur tensorflow.python.util.tf_exportrobjectrrr r r+s7! 56 -.;Jf;J/;Jr