Searched refs:ExecutePerReplica (Results 1 – 4 of 4) sorted by relevance
153 output_buffer_tup = executable.ExecutePerReplica(per_replica_args)826 def ExecutePerReplica(self, arguments=None): member in Executable891 return [out.to_py() for out in self.ExecutePerReplica(arguments)]
143 StatusOr<LocalShapedBufferTuple*> ExecutePerReplica(
275 %unignore xla::swig::LocalExecutable::ExecutePerReplica;
264 StatusOr<LocalShapedBufferTuple*> LocalExecutable::ExecutePerReplica( in ExecutePerReplica() function in xla::swig::LocalExecutable