Add TPUHardwareFeature in the tpu topology proto and populate it based on the...
Add TPUHardwareFeature in the tpu topology proto and populate it based on the tpu version (variant) during configuration. PiperOrigin-RevId: 427565848
Showing
- tensorflow/core/protobuf/tpu/topology.proto 22 additions, 1 deletiontensorflow/core/protobuf/tpu/topology.proto
- tensorflow/python/distribute/cluster_resolver/tpu/tpu_cluster_resolver.py 15 additions, 0 deletions...n/distribute/cluster_resolver/tpu/tpu_cluster_resolver.py
- tensorflow/python/distribute/cluster_resolver/tpu/tpu_cluster_resolver_test.py 13 additions, 0 deletions...tribute/cluster_resolver/tpu/tpu_cluster_resolver_test.py
- tensorflow/python/tpu/tpu_strategy_util.py 1 addition, 0 deletionstensorflow/python/tpu/tpu_strategy_util.py
- tensorflow/tools/api/golden/v1/tensorflow.distribute.cluster_resolver.-t-p-u-cluster-resolver.pbtxt 8 additions, 0 deletions...distribute.cluster_resolver.-t-p-u-cluster-resolver.pbtxt
- tensorflow/tools/api/golden/v2/tensorflow.distribute.cluster_resolver.-t-p-u-cluster-resolver.pbtxt 8 additions, 0 deletions...distribute.cluster_resolver.-t-p-u-cluster-resolver.pbtxt
Please register or sign in to comment