Searched refs:is_atomic_fn (Results 1 – 1 of 1) sorted by relevance
145 def map_structure_with_atomic(is_atomic_fn, map_fn, nested): argument161 if is_atomic_fn(nested):173 map_structure_with_atomic(is_atomic_fn, map_fn, ele) for ele in values