tf.tpu.experimental.HardwareFeature

class holds all the feature info about the TPU.

tpu_hardware_feature_protoprotobuf which describe the tpu hardware feature.

embedding_featureTPU embedding feature.
num_embedding_devices_per_chipNumber of embedding accelerator devices per chip.

Child Classes

class EmbeddingFeature