Searched refs:_per_replica_function (Results 1 – 1 of 1) sorted by relevance
955 def _per_replica_function(model): function965 _per_replica_function, args=(get_distributed_model(model, mode),))995 def _per_replica_function(model): function1012 _per_replica_function, args=(get_distributed_model(model, mode),))