TinyTorch/tests/regression/test_transformer_reshaping.py at dev

mirror of https://github.com/MLSysBook/TinyTorch.git synced 2026-04-27 20:07:31 -05:00

Files

Vijay Janapa Reddi a622e2c200 Fix regression tests for current API

- Update TransformerBlock to use mlp_ratio instead of hidden_dim
- Update PositionalEncoding argument order
- Fix MultiHeadAttention to use self-attention API
- Add missing MultiHeadAttention import

2025-12-02 22:30:42 -08:00

9.1 KiB

Raw Permalink Blame History

View Raw

9.1 KiB Raw Permalink Blame History

9.1 KiB

Raw Permalink Blame History