Searched refs:is_atomic_fn (Results 1 – 2 of 2) sorted by relevance
107 def map_structure_with_atomic(is_atomic_fn, map_fn, nested): argument123 if is_atomic_fn(nested):137 map_structure_with_atomic(is_atomic_fn, map_fn, ele) for ele in values
190 is_atomic_fn=lambda x: isinstance(x, int),