• vfdev's avatar
    Added CPU/CUDA and batch input for dtype conversion op (#2755) · fdca3073
    vfdev authored
    
    
    * 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: default avatarBrian <nairbv@yahoo.com>
    fdca3073
common_utils.py 15.2 KB