Skip to content

feat: map TFLite indices computed at run time - #130

Merged
asteinh merged 1 commit into
developfrom
feat/tflite-runtime-indices
Oct 4, 2026
Merged

asteinh merged 1 commit into
developfrom
feat/tflite-runtime-indices

Conversation

@asteinh

@asteinh asteinh commented Oct 4, 2026

Copy link
Copy Markdown
Member
  • GATHER, GATHER_ND, EMBEDDING_LOOKUP and DYNAMIC_UPDATE_SLICE take their indices or start positions from an int32 model input or an ARG_MAX/ARG_MIN output, besides constants.
  • A DYNAMIC_UPDATE_SLICE with run-time starts runs in TFLite's axis order, so the starts are used as they are.
  • Refused by name: an index tensor feeding an operand other than indices, run-time indices computed by any operator other than ARG_MAX/ARG_MIN, and int64 run-time indices.
  • Ten new corpus cases (gather, gather_nd, embedding_lookup, ArgMax feeding gather, and dynamic_update_slice with starts that reach past both ends), int8 and float32, match TFLite Micro byte for byte, untiled and at the 256-byte budget.

@asteinh
asteinh merged commit 7618f47 into develop Oct 4, 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