Searched refs:_per_replica_function (Results 1 – 1 of 1) sorted by relevance
941 def _per_replica_function(model): function951 _per_replica_function, args=(get_distributed_model(model, mode),))981 def _per_replica_function(model): function998 _per_replica_function, args=(get_distributed_model(model, mode),))