flashinfer.cake_rmsnorm_det_train.weight_dtypes¶
- flashinfer.cake_rmsnorm_det_train.weight_dtypes(arch: str, hidden: int, kind: str, residual: bool) tuple[str, ...]¶
The weight dtypes (tokens of
WEIGHT_DTYPES) the route ships onarch.
Back to top