Function repeat_kv

moe/model_config.py:148–155 · view source on GitHub ↗

torch.repeat_interleave(x, dim=2, repeats=n_rep).

(x: torch.Tensor, n_rep: int)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

forwardMethod · 0.85

no outgoing calls

no test coverage detected