Function op

k_diffusion/models/flops.py:35–37

(op, *args, **kwargs)

def op(op, *args, **kwargs):
    if getattr(state, "flop_counter", None):
        state.flop_counter.op(op, *args, **kwargs)


def op_linear(x, weight):