Home
last modified time | relevance | path

Searched defs:logits_tensor (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
H A Dlog_softmax.cc44 ITensorProxyPtr logits_tensor = inputs.at(0).tensor(); in Validate() local
64 ITensorProxyPtr logits_tensor = inputs.at(0).tensor(); in Convert() local
/aosp_15_r20/external/executorch/examples/qualcomm/oss_scripts/llama3_2/runner/
H A Drunner.cpp101 int32_t Runner::logitsToToken(const Tensor& logits_tensor) { in logitsToToken()
190 Tensor& logits_tensor = output_tensors.back()[0]; in generate() local
/aosp_15_r20/external/executorch/examples/qualcomm/qaihub_scripts/llama/runner/
H A Drunner.cpp135 int32_t Runner::logitsToToken(const Tensor& logits_tensor) { in logitsToToken()
286 Tensor& logits_tensor = output_tensors.back().back(); in generate() local
/aosp_15_r20/external/executorch/examples/qualcomm/oss_scripts/llama2/runner/
H A Drunner.cpp144 int32_t Runner::logitsToToken(const Tensor& logits_tensor) { in logitsToToken()
370 Tensor& logits_tensor = logits_res.get(); in generate() local
/aosp_15_r20/external/executorch/extension/llm/runner/
H A Dtext_decoder_runner.h70 const executorch::aten::Tensor& logits_tensor) { in logits_to_token()
H A Dtext_prefiller.cpp70 auto logits_tensor = in prefill() local
H A Dtext_token_generator.h87 executorch::aten::Tensor& logits_tensor = logits_res.get(); in generate() local
/aosp_15_r20/external/executorch/examples/llm_manual/
H A Dmain.cpp56 Tensor logits_tensor = logits_evalue.get()[0].toTensor(); in generate() local
/aosp_15_r20/external/executorch/examples/models/phi-3-mini/
H A Drunner.cpp76 uint64_t Runner::logits_to_token(const exec_aten::Tensor& logits_tensor) { in logits_to_token()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Drandom_ops.cc279 const TfLiteTensor* logits_tensor = GetInput(context, node, 0); in EvalMultinomial() local