Home
last modified time | relevance | path

Searched defs:tokenBatchSize (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/executorch/examples/mediatek/executor_runner/llama_runner/llm_helper/
H A Dmask_builder.cpp57 const size_t tokenBatchSize, in buildMask()
112 const size_t tokenBatchSize, in updateMask()
151 const size_t tokenBatchSize, in buildMask()
173 const size_t tokenBatchSize, in updateMask()
220 bool MaskBuilder::adjustMaskForPadding(const size_t tokenBatchSize) { in adjustMaskForPadding()
H A Drotary_embedding.cpp214 const size_t tokenBatchSize, in setEmbed()
252 const size_t tokenBatchSize, in setEmbed()
324 const size_t tokenBatchSize, in setEmbed()
/aosp_15_r20/external/executorch/examples/mediatek/executor_runner/llama_runner/
H A DLlamaModelChunk.cpp105 bool LlamaModelChunk::HotSwapModel(const size_t tokenBatchSize) { in HotSwapModel()
H A DModelChunk.cpp97 bool ModelChunk::HotSwapModel(const size_t tokenBatchSize) { in HotSwapModel()