tf.experimental.dtensor.run_on
Запускает заключённые в неё функции в области действия устройства DTensor. (устарела)
@contextlib.contextmanager
tf.experimental.dtensor.run_on(
mesh: tf.experimental.dtensor.Mesh
)
Эта функция возвращает область действия. Все операции и tf.functions в этой области будут выполняться на устройстве DTensor с использованием предоставленной сетки. Это полезно для обёртывания любой tf.function, которая не принимает DTensor в качестве входных данных, но должна возвращать DTensor. Эта область действия также гарантирует, что все небольшие константы будут реплицированы как DTensor.
| Аргументы | |
|---|---|
mesh | Экземпляр Mesh, из которого извлекается сетка по умолчанию. |
| Возвращает | |
|---|---|
| Контекст, в котором все операции и tf.functions будут выполняться на устройстве DTensor. |
© 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/experimental/dtensor/run_on
