Home
last modified time | relevance | path

Searched defs:merge_devices (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Ddevice_setter.py72 def __init__(self, ps_tasks, ps_device, worker_device, merge_devices, ps_ops, argument
134 merge_devices=True, argument
/external/pytorch/torch/_subclasses/
Dfake_tensor.py837 def merge_devices(t: object) -> None: function