Searched refs:struct_template (Results 1 – 2 of 2) sorted by relevance
138 def _Pack(elements, struct_template): argument152 return nest.pack_sequence_as(struct_template, [elements])153 return nest.pack_sequence_as(struct_template, elements)156 def _EmptyAcc(slen, struct_template): argument178 return nest.map_structure(_EmptyAccForTensor, struct_template)
41 def _SetShapeFromTemplate(struct, struct_template): argument43 template_as_list = nest.flatten(struct_template)