Home
last modified time | relevance | path

Searched refs:TextDecoderRunner (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/executorch/extension/llm/runner/
H A Dtext_decoder_runner.h23 class ET_EXPERIMENTAL TextDecoderRunner {
25 TextDecoderRunner(
31 virtual ~TextDecoderRunner() = default;
114 using ::executorch::extension::llm::TextDecoderRunner;
H A Dtext_decoder_runner.cpp24 TextDecoderRunner::TextDecoderRunner( in TextDecoderRunner() function in executorch::extension::llm::TextDecoderRunner
40 ::executorch::runtime::Result<exec_aten::Tensor> TextDecoderRunner::step( in step()
H A Dtext_prefiller.h25 TextDecoderRunner* text_decoder_runner,
41 TextDecoderRunner* text_decoder_runner_;
H A Dtext_token_generator.h25 TextDecoderRunner* text_decoder_runner, in TextTokenGenerator()
134 TextDecoderRunner* text_decoder_runner_;
H A Dtext_prefiller.cpp19 TextDecoderRunner* text_decoder_runner, in TextPrefiller()
H A Dmultimodal_runner.h127 std::unique_ptr<TextDecoderRunner> text_decoder_runner_;
/aosp_15_r20/external/executorch/examples/models/llava/runner/
H A Dllava_text_decoder_runner.h18 : public executorch::extension::llm::TextDecoderRunner {
24 : TextDecoderRunner(module, true, vocab_size, temperature){}; in LlavaTextDecoderRunner()
/aosp_15_r20/external/executorch/examples/models/llama/runner/
H A Drunner.h61 std::unique_ptr<::executorch::extension::llm::TextDecoderRunner>
H A Drunner.cpp123 text_decoder_runner_ = std::make_unique<llm::TextDecoderRunner>( in load()