Searched refs:_per_replica_function (Results 1 – 1 of 1) sorted by relevance
931 def _per_replica_function(model): function941 _per_replica_function, args=(get_distributed_model(model, mode),))971 def _per_replica_function(model): function988 _per_replica_function, args=(get_distributed_model(model, mode),))