Skip to content

Commit 7f63b11

Browse files
committed
update to_dtype_cvcuda name
1 parent 18f8c9c commit 7f63b11

File tree

1 file changed

+2
-2
lines changed
  • torchvision/transforms/v2/functional

1 file changed

+2
-2
lines changed

torchvision/transforms/v2/functional/_misc.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -369,7 +369,7 @@ def _to_dtype_tensor_dispatch(inpt: torch.Tensor, dtype: torch.dtype, scale: boo
369369
_cvcuda_to_torch_dtypes = {v: k for k, v in _torch_to_cvcuda_dtypes.items()}
370370

371371

372-
def _to_dtype_cvcuda(
372+
def _to_dtype_image_cvcuda(
373373
inpt: "cvcuda.Tensor",
374374
dtype: torch.dtype,
375375
scale: bool = False,
@@ -428,7 +428,7 @@ def _to_dtype_cvcuda(
428428

429429

430430
if CVCUDA_AVAILABLE:
431-
_register_kernel_internal(to_dtype, _import_cvcuda().Tensor)(_to_dtype_cvcuda)
431+
_register_kernel_internal(to_dtype, _import_cvcuda().Tensor)(_to_dtype_image_cvcuda)
432432

433433

434434
def sanitize_bounding_boxes(

0 commit comments

Comments
 (0)