Searched refs:_create_ta (Results 1 – 2 of 2) sorted by relevance
367 def _create_ta(s, d): function374 initial_outputs_ta = nest.map_structure(_create_ta, decoder.output_size,
779 def _create_ta(name, element_shape, dtype): function788 _create_ta(796 _create_ta(