Searched defs:logits_ (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/executorch/examples/qualcomm/qaihub_scripts/llama/runner/ | ||
| H A D | io_memory.h | 90 std::unique_ptr<executorch::aten::TensorImpl> logits_; variable |
| /aosp_15_r20/external/executorch/examples/qualcomm/oss_scripts/llama3_2/runner/ | ||
| H A D | io_memory.h | 98 std::unique_ptr<executorch::aten::TensorImpl> logits_; variable |
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ | ||
| H A D | multinomial_test.cc | 66 int logits_; member in tflite::__anon752fffe90111::MultinomialOpModel |
| H A D | random_ops_test.cc | 96 int logits_; member in tflite::__anon9905f3300111::MultinomialOpModel |
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ | ||
| H A D | softmax_op_gpu.cu.cc | 176 const T* logits_; member |