def METHOD_NAME(self, queries, bsize=None, to_cpu=False, context=None):
    if bsize:
        batches = self.query_tokenizer.tensorize(queries, context=context, bsize=bsize)
        batches = [self.query(input_ids, attention_mask, to_cpu=to_cpu) for input_ids, attention_mask in batches]
        return torch.cat(batches)
    input_ids, attention_mask = self.query_tokenizer.tensorize(queries, context=context)
    return self.query(input_ids, attention_mask)