def METHOD_NAME(
    self,
    x: torch.Tensor,
    x_len: torch.LongTensor,
    class_id: torch.LongTensor,
    unique_name: List[str],
    _dump_dir: str = None,
):
    spk_embeddings = self.predict(x, x_len)
    loss, logits = self.loss(spk_embeddings, class_id)
    prediction = [index for index in logits.argmax(dim=-1).detach().cpu().tolist()]
    cacheable = dict(
        loss=loss.detach().cpu().item(),
        class_id=class_id.detach().cpu().tolist(),
        prediction=prediction,
        unique_name=unique_name,
    )
    return loss, cacheable