Home
last modified time | relevance | path

Searched defs:logits_offset (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dmultinomial.cc101 const TfLiteTensor* logits, int logits_offset, in MultinomialSample()
185 int logits_offset = logits_size * batch; in Eval() local
/aosp_15_r20/external/executorch/examples/qualcomm/qaihub_scripts/llama/runner/
H A Drunner.cpp57 const int logits_offset) in Runner()