TensorRT-LLMs/tensorrt_llm/layers
Yuening Li 1f8ae2b2db
[TRTLLM-5863][feat] Support MoE INT8 Weight-Only-Quantization in PyTorch Workflow (#6629)
Signed-off-by: Yuening Li <62227368+yueningl@users.noreply.github.com>
2025-08-15 17:15:49 -04:00
..
__init__.py Update TensorRT-LLM (#2873) 2025-03-11 21:13:42 +08:00
activation.py Update TensorRT-LLM (#787) 2024-01-02 17:54:32 +08:00
attention.py [https://nvbugs/5302040][feat] Add whisper support (Bert Attention on SM100 and GPTAttention for cross attention on SM100) (#5527) 2025-08-13 11:19:13 -07:00
cast.py Update TensorRT-LLM (#787) 2024-01-02 17:54:32 +08:00
conv.py Update TensorRT-LLM (#2873) 2025-03-11 21:13:42 +08:00
embedding.py fix: #3137 speculative decoding and multimodal input support (#3276) 2025-04-09 23:40:19 +08:00
language_adapter.py Update TensorRT-LLM (#2849) 2025-03-04 18:44:00 +08:00
linear.py Update TensorRT-LLM (#2873) 2025-03-11 21:13:42 +08:00
lora.py Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
mlp.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
moe.py [TRTLLM-5863][feat] Support MoE INT8 Weight-Only-Quantization in PyTorch Workflow (#6629) 2025-08-15 17:15:49 -04:00
normalization.py Update TensorRT-LLM (#2849) 2025-03-04 18:44:00 +08:00
pooling.py Update TensorRT-LLM (#787) 2024-01-02 17:54:32 +08:00
recurrent.py Update TensorRT-LLM (#1954) 2024-07-16 15:30:25 +08:00
ssm.py Update TensorRT-LLM (#2562) 2024-12-11 00:31:05 -08:00