Home
last modified time | relevance | path

Searched defs:precompute_freqs_cis (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/executorch/examples/models/llama/
H A Drope.py43 def precompute_freqs_cis( function
/aosp_15_r20/external/pytorch/benchmarks/gpt_fast/
H A Dmodel.py269 def precompute_freqs_cis(seq_len: int, n_elem: int, base: int = 10000) -> Tensor: function
H A Dmixtral_moe_model.py282 def precompute_freqs_cis(seq_len: int, n_elem: int, base: int = 10000) -> Tensor: function