tf.distribute.experimental.CommunicationImplementation

Реализация межпроцессорной связи.

Основные псевдонимы

tf.distribute.experimental.CollectiveCommunication

Псевдонимы совместимости для миграции

См. Руководство по миграции для получения дополнительных сведений.

AUTO: Автоматически выбирается TensorFlow.
RING: Алгоритмы колец TensorFlow для all-reduce и all-gather.
NCCL: Библиотека NVIDIA® NCCL. В настоящее время используется только для all-reduce на GPU; all-reduce на CPU, all-gather и broadcast используют метод RING.

Переменные класса
AUTO	`<CommunicationImplementation.AUTO: 'AUTO'>`
NCCL	`<CommunicationImplementation.NCCL: 'NCCL'>`
RING	`<CommunicationImplementation.RING: 'RING'>`

© 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/versions/r2.9/api_docs/python/tf/distribute/experimental/CommunicationImplementation