Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Ddecomposition.py36 needs_fallback_due_to_atomic_add_limitations,
889 if reduction_type == "mean" and not needs_fallback_due_to_atomic_add_limitations(
H A Dutils.py1809 def needs_fallback_due_to_atomic_add_limitations(dtype): function
1838 and needs_fallback_due_to_atomic_add_limitations(src_dtype)
H A Dlowering.py67 needs_fallback_due_to_atomic_add_limitations,
3223 if accumulate and needs_fallback_due_to_atomic_add_limitations(self.get_dtype()):