openfold.utils.kernel.attention_core

Classes

AttentionCoreFunction(*args, **kwargs)

class AttentionCoreFunction(*args, **kwargs)

Bases: Function

static backward(ctx, grad_output)

static forward(ctx, q, k, v, bias_1=None, bias_2=None)