tf.experimental.dtensor.default_mesh

Устанавливает стандартную сетку устройств DTensor для использования в заключённых функциях.

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

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