Home
last modified time | relevance | path

Searched refs:last_step_tensor_outputs (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy.py360 last_step_tensor_outputs = [
372 output_num = len(last_step_tensor_outputs) // self._num_replicas_in_sync
373 last_step_tensor_outputs = [
374 last_step_tensor_outputs[i::output_num] for i in range(output_num)
378 last_step_tensor_outputs = []
380 _set_last_step_outputs(ctx, last_step_tensor_outputs)
655 def _set_last_step_outputs(ctx, last_step_tensor_outputs): argument
660 ctx.last_step_outputs, last_step_tensor_outputs)
Done_device_strategy.py134 last_step_tensor_outputs = loop_result[1:]
136 ctx.last_step_outputs, last_step_tensor_outputs)
Dmirrored_strategy.py630 last_step_tensor_outputs = loop_result[1:]
632 ctx.last_step_outputs, last_step_tensor_outputs)