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.py454 last_step_tensor_outputs = [
466 output_num = len(last_step_tensor_outputs) // self._num_replicas_in_sync
467 last_step_tensor_outputs = [
468 last_step_tensor_outputs[i::output_num] for i in range(output_num)
472 last_step_tensor_outputs = []
474 _set_last_step_outputs(ctx, last_step_tensor_outputs)
879 def _set_last_step_outputs(ctx, last_step_tensor_outputs): argument
884 ctx.last_step_outputs, last_step_tensor_outputs)
Done_device_strategy.py344 last_step_tensor_outputs = loop_result[1:]
346 ctx.last_step_outputs, last_step_tensor_outputs)
Dmirrored_strategy.py703 last_step_tensor_outputs = loop_result[1:]
705 ctx.last_step_outputs, last_step_tensor_outputs)