Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Ddecomposition.py958 torch._inductor.lowering.should_fallback_max_pool2d_with_indices(
H A Dlowering.py3774 def should_fallback_max_pool2d_with_indices(kernel_size, dilation): function
3802 use_fallback = should_fallback_max_pool2d_with_indices(kernel_size, dilation)
4308 if should_fallback_max_pool2d_with_indices(kernel_size, dilation=[1, 1]):