Function op_natten

k_diffusion/models/flops.py:51–54 · view source on GitHub ↗

(q, k, v, kernel_size)

Source from the content-addressed store, hash-verified

49
50
51	def op_natten(q, k, v, kernel_size):
52	*q_rest, d_q = q
53	*_, d_v = v
54	return math.prod(q_rest) * (d_q + d_v) * kernel_size**2

nothing calls this directly

no outgoing calls

no test coverage detected