Searched refs:map_structure_with_atomic (Results 1 – 2 of 2) sorted by relevance
107 def map_structure_with_atomic(is_atomic_fn, map_fn, nested): function137 map_structure_with_atomic(is_atomic_fn, map_fn, ele) for ele in values197 return map_structure_with_atomic(_is_atomic_shape, _convert_shape,252 return map_structure_with_atomic(_is_atomic_nested, _convert_object_or_list,
189 tf_utils.map_structure_with_atomic(