Searched refs:map_structure_with_atomic (Results 1 – 1 of 1) sorted by relevance
106 def map_structure_with_atomic(is_atomic_fn, map_fn, nested): function136 map_structure_with_atomic(is_atomic_fn, map_fn, ele) for ele in values196 return map_structure_with_atomic(_is_atomic_shape, _convert_shape,251 return map_structure_with_atomic(_is_atomic_nested, _convert_object_or_list,