Searched refs:per_replica_args (Results 1 – 1 of 1) sorted by relevance
109 def execute_replicated(self, executable, per_replica_args): argument152 def execute_replicated(self, executable, per_replica_args): argument153 output_buffer_tup = executable.ExecutePerReplica(per_replica_args)196 def execute_replicated(self, executable, per_replica_args): argument197 if len(per_replica_args) != 1:200 return [executable.Execute(per_replica_args[0])]