needs_fallback_due_to_atomic_add_limitations (reference) in projects: aosp_15_r20

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 D	decomposition.py	`36 needs_fallback_due_to_atomic_add_limitations, 889 if reduction_type == "mean" and not needs_fallback_due_to_atomic_add_limitations(`
H A D	utils.py	`1809 def needs_fallback_due_to_atomic_add_limitations(dtype): function 1838 and needs_fallback_due_to_atomic_add_limitations(src_dtype)`
H A D	lowering.py	`67 needs_fallback_due_to_atomic_add_limitations, 3223 if accumulate and needs_fallback_due_to_atomic_add_limitations(self.get_dtype()):`