Searched refs:include_tail (Results 1 – 4 of 4) sorted by relevance
| /external/executorch/examples/mediatek/models/llm_models/ |
| D | modeling_base.py | 13 include_tail=False, argument
23 self.include_tail = include_tail
|
| D | modeling_llama.py | 64 include_tail=False, argument
73 include_tail,
|
| D | modeling_common.py | 276 include_tail=False, argument
282 config, num_blocks, chunk_idx, dtype, include_tail, return_attn, jit_trace
295 if self.include_tail:
345 if self.include_tail:
656 if self.include_tail:
718 if self.include_tail:
|
| /external/executorch/examples/mediatek/model_export_scripts/ |
| D | llama.py | 434 include_tail=(chunk_idx == args.num_chunks - 1),
|