Searched refs:map_structure_with_atomic (Results 1 – 2 of 2) 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,
185 tf_utils.map_structure_with_atomic(