Skip to content

[Test][TFLite] Add unit tests for PRELU#19402

Merged
tlopex merged 5 commits intoapache:mainfrom
fnhirwa:prelu
Apr 14, 2026
Merged

[Test][TFLite] Add unit tests for PRELU#19402
tlopex merged 5 commits intoapache:mainfrom
fnhirwa:prelu

Conversation

@fnhirwa
Copy link
Copy Markdown
Contributor

@fnhirwa fnhirwa commented Apr 14, 2026

This PR adds unit test coverage for PRELU activation
in the Relax TFLite frontend, as part of #18971

  • Added unit test for PRELU and

Enabled converter to handle alpha broadcasting more cleanly across constant and expression-backed alpha inputs.

Copy link
Copy Markdown
Contributor

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request refactors the PReLU conversion in the TFLite frontend to handle alpha tensor broadcasting and adds new unit tests for structural and runtime verification. The review feedback highlights an opportunity to further optimize the implementation by using get_tensor_expr to unify the logic for constant and expression-backed tensors, thereby avoiding IR bloat caused by materializing large broadcasted constants.

Copy link
Copy Markdown
Member

@tlopex tlopex left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@tlopex tlopex merged commit 5efa4b7 into apache:main Apr 14, 2026
9 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.

2 participants