Searched refs:needs_fallback_due_to_atomic_add_limitations (Results 1 – 3 of 3) sorted by relevance
36 needs_fallback_due_to_atomic_add_limitations,889 if reduction_type == "mean" and not needs_fallback_due_to_atomic_add_limitations(
1809 def needs_fallback_due_to_atomic_add_limitations(dtype): function1838 and needs_fallback_due_to_atomic_add_limitations(src_dtype)
67 needs_fallback_due_to_atomic_add_limitations,3223 if accumulate and needs_fallback_due_to_atomic_add_limitations(self.get_dtype()):