Skip to content

feat: tile isolated reshapes with integral row mappings - #133

Merged
asteinh merged 1 commit into
developfrom
feat/tile-reshape
Oct 5, 2026
Merged

asteinh merged 1 commit into
developfrom
feat/tile-reshape

Conversation

@asteinh

@asteinh asteinh commented Oct 5, 2026

Copy link
Copy Markdown
Member
  • The partitioner tiles a single-op Reshape stage when the leading block count is equal on both sides and every band, the last one included, maps to whole output rows. Band sizes are multiples of Co / gcd(Ci, Co). Matrix-row axes are tried first, then stored axis 1.
  • A Reshape that changes the block count, reorders elements or sits inside a chain stays untiled.
  • 14 of the 18 Reshape stages in the TFLite corpus that exceeded 256 bytes now tile there. 7 of the 12 affected models compile at 256 bytes and match TFLite Micro byte for byte.

Needs the runtime branch of the same name.

@asteinh
asteinh merged commit eb15b86 into develop Oct 5, 2026
15 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant