Searched refs:_per_device_function (Results 1 – 1 of 1) sorted by relevance
768 def _per_device_function(model): function778 _per_device_function, args=(get_distributed_model(model, mode),))808 def _per_device_function(model): function825 _per_device_function, args=(get_distributed_model(model, mode),))