Lines Matching refs:ta
761 input_ta = tuple(ta.unstack(input_)
762 for ta, input_ in zip(input_ta, flat_input))
781 input_t = tuple(ta.read(time) for ta in input_ta)
786 input_t = tuple(ta[time.numpy()] for ta in input_ta)
810 ta.write(time, out) for ta, out in zip(output_ta_t, output))
812 for ta, out in zip(output_ta_t, output):
813 ta[time.numpy()] = out
836 final_outputs = tuple(ta.stack() for ta in output_final_ta)
1145 lambda ta, emit: ta.write(time, emit), emit_ta, emit_output)