Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_inductor/kernel/
H A Dconv.py570 if torch._inductor.utils._use_conv_autotune_backend("ATEN"):
581 torch._inductor.utils._use_conv_autotune_backend("TRITON")
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dutils.py1075 def _use_conv_autotune_backend(backend: str) -> bool: function