tf.contrib.framework.VariableDeviceChooser

Выбор устройства для переменных.

tf.contrib.framework.VariableDeviceChooser(
    num_tasks=0, job_name='ps', device_type='CPU', device_index=0, replica=None
)

При использовании параметрического сервера переменные будут распределяться по круговому циклу. В противном случае это позволяет размещение на GPU или CPU.

Аргументы
`num_tasks`	количество задач.
`job_name`	Строка, имя задания параметрического сервера.
`device_type`	Необязательная строка типа устройства (например, "CPU" или "GPU")
`device_index`	целое число. Необязательный индекс устройства. Если не указано, устройство представляет собой любой device_index.

Методы

`call`

Просмотреть исходный код

__call__(
    op
)

Вызов self как функции.

© 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/r1.15/api_docs/python/tf/contrib/framework/VariableDeviceChooser