Searched refs:should_fallback_max_pool2d_with_indices (Results 1 – 2 of 2) sorted by relevance
958 torch._inductor.lowering.should_fallback_max_pool2d_with_indices(
3774 def should_fallback_max_pool2d_with_indices(kernel_size, dilation): function3802 use_fallback = should_fallback_max_pool2d_with_indices(kernel_size, dilation)4308 if should_fallback_max_pool2d_with_indices(kernel_size, dilation=[1, 1]):