Exposes tf.nest.{flatten,map_structure,pack_sequence_as,assert_same_structure}