tf.tpu.experimental.HardwareFeature
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. |
num_embedding_devices_per_chip | Number of embedding accelerator devices per chip. |
Child Classes
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/tpu/experimental/HardwareFeature
