feat: ensure i32 indexing based on input sizes

[avik-pal]

No description provided.

[codecov]

Codecov Report

❌ Patch coverage is 80.00000% with 2 lines in your changes missing coverage. Please review.
✅ Project coverage is 42.58%. Comparing base (678c3b5) to head (8db700b).
⚠️ Report is 4 commits behind head on main.

Files with missing lines	Patch %	Lines
src/Ops.jl	80.00%	2 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1630      +/-   ##
==========================================
+ Coverage   42.57%   42.58%   +0.01%     
==========================================
  Files         123      123              
  Lines       21901    21909       +8     
==========================================
+ Hits         9324     9330       +6     
- Misses      12577    12579       +2     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.