multinomial_sample_one_no_sync (reference) in projects: aosp_15_r20

Searched refs:multinomial_sample_one_no_sync (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/executorch/examples/models/llama/experimental/
H A D	generate.py	`18 def multinomial_sample_one_no_sync( function 38 idx_next = multinomial_sample_one_no_sync(probs)`
/aosp_15_r20/external/pytorch/benchmarks/gpt_fast/
H A D	generate.py	`53 def multinomial_sample_one_no_sync( function 73 idx_next = multinomial_sample_one_no_sync(probs)`