Searched refs:last_step_tensor_outputs (Results 1 – 3 of 3) sorted by relevance
1121 last_step_tensor_outputs = [1133 output_num = len(last_step_tensor_outputs) // self._num_replicas_in_sync1134 last_step_tensor_outputs = [1135 last_step_tensor_outputs[i::output_num] for i in range(output_num)1139 last_step_tensor_outputs = []1141 _set_last_step_outputs(ctx, last_step_tensor_outputs)1827 def _set_last_step_outputs(ctx, last_step_tensor_outputs): argument1832 ctx.last_step_outputs, last_step_tensor_outputs)
384 last_step_tensor_outputs = loop_result[1:]386 ctx.last_step_outputs, last_step_tensor_outputs)
644 last_step_tensor_outputs = loop_result[1:]646 ctx.last_step_outputs, last_step_tensor_outputs)