tf.distribute.experimental.CollectiveCommunication
| Просмотреть исходный код на GitHub |
Варианты коммуникации для CollectiveOps.
-
AUTO: По умолчанию использует автоматические настройки среды выполнения. -
RING: Алгоритмы колец TensorFlow для all-reduce и all-gather. -
NCCL: Использует ncclAllReduce для all-reduce и алгоритмы колец для all-gather.
Переменные класса
-
AUTO -
NCCL -
RING
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/distribute/experimental/CollectiveCommunication
