mirror of
https://github.com/MLSysBook/TinyTorch.git
synced 2026-03-09 13:42:01 -05:00
- test_spatial_gradient_flow.py: Tests Conv2d and MaxPool2d backward function attachment and gradient propagation - test_embedding_gradient_flow.py: Tests Embedding backward function attachment and gradient propagation - Tests verify _grad_fn attachment to prevent .data bypass issues - Tests validate gradient flow to all parameters (weight, bias) - Tests check end-to-end gradient chains - All tests pass (8/8 spatial, 6/6 embedding)