Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Eltwise mul for different batch size #6587

Merged
merged 2 commits into from
Mar 22, 2024
Merged

Conversation

umadevimcw
Copy link
Contributor

  • Used the repeat op to broadcast the data for different batch sizes and perform binary eltwise mul operation
  • Updated the tensor in commonplace. Now it is working for all binary op with different batch size
  • Added the test case for the same

@umadevimcw
Copy link
Contributor Author

@tt-aho @eyonland and @kpaigwar Please share your comments.
Current logic support only batch support

@umadevimcw umadevimcw requested a review from arakhmati March 20, 2024 11:25
@umadevimcw umadevimcw force-pushed the umadevimcw/batchmul_variant2 branch from e2c02c3 to fca3190 Compare March 22, 2024 06:16
@umadevimcw umadevimcw force-pushed the umadevimcw/batchmul_variant2 branch from fca3190 to 87a7ae0 Compare March 22, 2024 12:53
@umadevimcw
Copy link
Contributor Author

@umadevimcw umadevimcw merged commit 9657ed3 into main Mar 22, 2024
4 checks passed
@umadevimcw umadevimcw deleted the umadevimcw/batchmul_variant2 branch March 25, 2024 07:13
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.

3 participants