Searched refs:_execute_replicated (Results 1 – 1 of 1) sorted by relevance
731 return partial(_execute_replicated, compiled, out_avals, result_handlers)845 def _execute_replicated(compiled: XlaExecutable, avals, handlers, *args): function