tf.experimental.dtensor.run_on

Выполняет заключённые функции в области устройства DTensor. (устарело)

@contextlib.contextmanager
tf.experimental.dtensor.run_on(
    mesh: tf.experimental.dtensor.Mesh
)

Эта функция возвращает область. Все операции и tf.функции в этой области будут выполняться на устройстве DTensor с использованием предоставленной сетки. Это полезно для обертывания любого tf.function, который не принимает DTensor в качестве входных данных, но хотел бы получить DTensor в качестве результата. Область также будет гарантировать, что все небольшие константы будут дублированы как DTensor.