View source on GitHub |
class holds all the feature info about the TPU.
tf.tpu.experimental.HardwareFeature(
tpu_hardware_feature_proto
)
Args | |
---|---|
tpu_hardware_feature_proto
|
protobuf which describe the tpu hardware feature. |
Attributes | |
---|---|
embedding_feature
|
TPU embedding feature. |