Searched refs:replica_args (Results 1 – 2 of 2) sorted by relevance
841 arguments = [list(replica_args) for replica_args in arguments]844 for replica, replica_args in enumerate(arguments):845 for arg in replica_args:856 [arg.c_buffer for arg in replica_args] for replica_args in arguments889 put(arg, self._device_ordinals[replica]) for arg in replica_args890 ] for replica, replica_args in enumerate(arguments)]
167 def replicated_fn(replica_id, replica_args, replica_kwargs): argument170 result[0] = fn(*replica_args, **replica_kwargs)