-
Notifications
You must be signed in to change notification settings - Fork 7.1k
Added CPU/CUDA and batch input for dtype conversion op #2755
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
Added CPU/CUDA and batch input for dtype conversion op #2755
Conversation
Codecov Report
@@ Coverage Diff @@
## master #2755 +/- ##
=======================================
Coverage 73.08% 73.08%
=======================================
Files 96 96
Lines 8319 8319
Branches 1293 1293
=======================================
Hits 6080 6080
Misses 1840 1840
Partials 399 399 Continue to review full report at Codecov.
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
#2485 has been merged, should we rebase on top of master now?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks!
* make convert_image_dtype scriptable * move convert dtype to functional_tensor since only works on tensors * retain availability of convert_image_dtype in functional.py * Update code and tests * Replaced int by torch.dtype * int -> torch.dtype and use F instead of F_t * Update functional_tensor.py * Added CPU/CUDA+batch tests * Fixed tests according to review Co-authored-by: Brian <[email protected]>
* make convert_image_dtype scriptable * move convert dtype to functional_tensor since only works on tensors * retain availability of convert_image_dtype in functional.py * Update code and tests * Replaced int by torch.dtype * int -> torch.dtype and use F instead of F_t * Update functional_tensor.py * Added CPU/CUDA+batch tests * Fixed tests according to review Co-authored-by: Brian <[email protected]>
Description:
ConvertImageDtype
Blocked by #2485