frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[refactor] is_known_channels_last_contiguous* -> definitely_channels_last_contiguous* (#155499)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/155499
Approved by: https://github.com/laithsakka
This commit is contained in:
Colin Peppler
2025-06-09 16:12:26 -07:00
committed by PyTorch MergeBot
parent 94763f5ca7
commit c9404faacb
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
torch/_prims_common/__init__.py
View File

@ -396,12 +396,12 @@ def definitely_contiguous(a: TensorLikeType) -> bool:
# similar to is_channels_last_contiguous_2d but return false on data dependency.
def is_known_channels_last_contiguous_2d(a: Tensor) -> bool:
def definitely_channels_last_contiguous_2d(a: Tensor) -> bool:
return is_channels_last_contiguous_2d(a, false_if_dde=True)
# similar to is_channels_last_contiguous_3d but return false on data dependency.
def is_known_channels_last_contiguous_3d(a: Tensor) -> bool:
def definitely_channels_last_contiguous_3d(a: Tensor) -> bool:
return is_channels_last_contiguous_3d(a, false_if_dde=True)
@ -432,10 +432,10 @@ def is_channels_last_contiguous(a: Tensor) -> bool:
# similar to is_channels_last_contiguous but return false on data dependency.
def is_known_channels_last_contiguous(a: Tensor) -> bool:
return is_known_channels_last_contiguous_2d(
def definitely_channels_last_contiguous(a: Tensor) -> bool:
return definitely_channels_last_contiguous_2d(
a
) or is_known_channels_last_contiguous_3d(a)
) or definitely_channels_last_contiguous_3d(a)
def is_non_overlapping_and_dense(a: Tensor) -> bool: