cross_attn

core.nn.cross_attn(module, *, x_query, x_context, query_pos_emb, context_pos_emb, query_seqlen, context_seqlen, impl)

forward_varlen for RotaryCrossAttention, on either backend.

Return type:

Tensor