Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributions/
H A Dutils.py82 def logits_to_probs(logits, is_binary=False): function
/aosp_15_r20/external/executorch/examples/models/llama/experimental/
H A Dgenerate.py25 def logits_to_probs(logits, temperature: float = 1.0, top_k: Optional[int] = None): function
/aosp_15_r20/external/pytorch/benchmarks/gpt_fast/
H A Dgenerate.py60 def logits_to_probs(logits, temperature: float = 1.0, top_k: Optional[int] = None): function