Skip to content

[CMSIS-NN] Removed batch_size assert in depthwise_conv#3562

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

[CMSIS-NN] Removed batch_size assert in depthwise_conv#3562
veblush wants to merge 1 commit into
tensorflow:mainfrom
veblush:cm-dw

Conversation

@veblush
Copy link
Copy Markdown
Collaborator

@veblush veblush commented May 21, 2026

Since CMSIS-NN actually supports batch sizes > 1 natively (it loops over input_batches internally), this assertion seems to be outdated or unnecessarily restrictive. Therefore, I removed the batch_size == 1 check.

BUG=N/A

@veblush veblush requested a review from a team as a code owner May 21, 2026 18:06
@veblush veblush changed the title Removed batch_size assert in depthwise_conv [CMSIS-NN Removed batch_size assert in depthwise_conv May 21, 2026
@veblush veblush changed the title [CMSIS-NN Removed batch_size assert in depthwise_conv [CMSIS-NN] Removed batch_size assert in depthwise_conv May 21, 2026
@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