Skip to content

[CMSIS-NN] Fixed rank support in pad and min/max#3563

Open
veblush wants to merge 1 commit into
tensorflow:mainfrom
veblush:cm-pad
Open

[CMSIS-NN] Fixed rank support in pad and min/max#3563
veblush wants to merge 1 commit into
tensorflow:mainfrom
veblush:cm-pad

Conversation

@veblush
Copy link
Copy Markdown
Collaborator

@veblush veblush commented May 21, 2026

  • maximum_minimum.cc: Updated FillVariableShape to support rank 1 inputs (adding {1, 1, 1, tensor_dims[0]}).
  • pad.cc: Added conditional logic to properly handle shapes of rank 1, 2, and 3 by converting sizes and paddings correctly into 4D cmsis_nn_dims structures.

BUG=N/A

@veblush veblush requested a review from a team as a code owner May 21, 2026 18:11
@veblush veblush added the ci:full Triggers the comprehensive cross-platform test suite. label May 21, 2026
@veblush veblush requested a review from suleshahid May 21, 2026 21:42
@veblush veblush added ci:full Triggers the comprehensive cross-platform test suite. and removed ci:full Triggers the comprehensive cross-platform test suite. labels May 21, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ci:full Triggers the comprehensive cross-platform test suite.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants