Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dir.py2813 def stride_ordered_for_memory_format(sizes, memory_format): member in FlexibleLayout
H A Dlowering.py6190 out_stride = ir.FlexibleLayout.stride_ordered_for_memory_format(size, memory_format)