Searched refs:old_ta (Results 1 – 2 of 2) sorted by relevance
1245 def build_ta_with_new_flow(old_ta, flow): argument1248 dtype=old_ta.dtype,1249 dynamic_size=old_ta._dynamic_size,1250 handle=old_ta.handle,1252 infer_shape=old_ta._infer_shape,1253 colocate_with_first_write_call=old_ta._colocate_with_first_write_call)1254 ta._colocate_with = old_ta._colocate_with1255 ta._element_shape = old_ta._element_shape
826 old_ta=flattened_structure[i], flow=flat_sequence[i])