Home
last modified time | relevance | path

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

/external/executorch/examples/mediatek/executor_runner/
Dmtk_llama_executor_runner.cpp223 const auto logits_type = llama_runtime.GetModelOptions().model_output_type; in digest_prompt() local
254 const auto logits_type = llama_runtime.GetModelOptions().model_output_type; in gen_response() local
Dmtk_llama_runner.cpp232 const auto logits_type = llama_runtime.GetModelOptions().model_output_type; in digest_prompt() local
264 const auto logits_type = llama_runtime.GetModelOptions().model_output_type; in gen_response() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsoftmax_op.cc151 const DataType logits_type = input_type(0); in Compile() local