Home
last modified time | relevance | path

Searched refs:is_contiguous_for_memory_format (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/nested/_internal/
H A Dops.py405 from torch._prims_common import is_contiguous_for_memory_format
421 return is_contiguous_for_memory_format(inp._values, **new_kwargs)
/aosp_15_r20/external/pytorch/torch/_prims_common/
H A D__init__.py310 def is_contiguous_for_memory_format( # type: ignore[return] function
/aosp_15_r20/external/pytorch/torch/_refs/
H A D__init__.py1991 or utils.is_contiguous_for_memory_format(a, memory_format=memory_format)
2946 if utils.is_contiguous_for_memory_format(a, memory_format=memory_format):