Searched refs:map_structure_with_atomic (Results 1 – 1 of 1) sorted by relevance
145 def map_structure_with_atomic(is_atomic_fn, map_fn, nested): function173 map_structure_with_atomic(is_atomic_fn, map_fn, ele) for ele in values224 return map_structure_with_atomic(_is_atomic_shape, _convert_shape,272 return map_structure_with_atomic(_is_atomic_nested, _convert_object_or_list,