Home
last modified time | relevance | path

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

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