Method op

k_diffusion/models/flops.py:24–25

(self, op, *args, **kwargs)

        self.ops = []

    def op(self, op, *args, **kwargs):
        self.ops.append((op, args, kwargs))

    @property
    def flops(self):