View source on GitHub |
Please see the definition of these values in TPUConfig.
Migrate to TF2
TPU Estimator manages its own TensorFlow graph and session, so it is not
compatible with TF2 behaviors. We recommend that you migrate to the newer
tf.distribute.TPUStrategy
. See the
TPU guide for details.
Description
Class Variables | |
---|---|
BROADCAST |
4
|
PER_HOST_V1 |
2
|
PER_HOST_V2 |
3
|
PER_SHARD_V1 |
1
|
SLICED |
5
|