Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DSerialStackImpl.h81 auto first_tensor_mem_format = first_tensor.suggest_memory_format(); in can_use_native_serial_stack_impl() local
84 if (!result.is_contiguous(first_tensor_mem_format)) { in can_use_native_serial_stack_impl()
106 if (!tensor.is_contiguous(first_tensor_mem_format) || in can_use_native_serial_stack_impl()