Spec-Zone.ru › TensorFlow

tf.raw_ops.TPUOrdinalSelector

Операция выбора ядра TPU.

Просмотреть псевдонимы

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

Подробнее см. в руководстве по миграции.

tf.compat.v1.raw_ops.TPUOrdinalSelector

tf.raw_ops.TPUOrdinalSelector(
    name=None
)

Эта операция формирует набор ядер TPU (для прогрева) или одно ядро TPU (для обычного выполнения инференса), на котором будет выполняться программа TPU. Результат используется операцией TPUPartitionedCall.

Аргументы
name Имя операции (необязательно).
Возвращает
Tensor типа int32.

© 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/raw_ops/TPUOrdinalSelector

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API