Searched defs:logits_tensor (Results 1 – 10 of 10) sorted by relevance
44 ITensorProxyPtr logits_tensor = inputs.at(0).tensor(); in Validate() local64 ITensorProxyPtr logits_tensor = inputs.at(0).tensor(); in Convert() local
70 const executorch::aten::Tensor& logits_tensor) { in logits_to_token()
70 auto logits_tensor = in prefill() local
87 executorch::aten::Tensor& logits_tensor = logits_res.get(); in generate() local
101 int32_t Runner::logitsToToken(const Tensor& logits_tensor) { in logitsToToken()190 Tensor& logits_tensor = output_tensors.back()[0]; in generate() local
135 int32_t Runner::logitsToToken(const Tensor& logits_tensor) { in logitsToToken()286 Tensor& logits_tensor = output_tensors.back().back(); in generate() local
56 Tensor logits_tensor = logits_evalue.get()[0].toTensor(); in generate() local
76 uint64_t Runner::logits_to_token(const exec_aten::Tensor& logits_tensor) { in logits_to_token()
144 int32_t Runner::logitsToToken(const Tensor& logits_tensor) { in logitsToToken()370 Tensor& logits_tensor = logits_res.get(); in generate() local
279 const TfLiteTensor* logits_tensor = GetInput(context, node, 0); in EvalMultinomial() local