tf.tpu.experimental.HardwareFeature

class holds all the feature info about the TPU.

tpu_hardware_feature_proto protobuf which describe the tpu hardware feature.

embedding_feature TPU embedding feature.
num_embedding_devices_per_chip Number of embedding accelerator devices per chip.

Child Classes

class EmbeddingFeature