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.py1063 last_step_tensor_outputs = [
1075 output_num = len(last_step_tensor_outputs) // self._num_replicas_in_sync
1076 last_step_tensor_outputs = [
1077 last_step_tensor_outputs[i::output_num] for i in range(output_num)
1081 last_step_tensor_outputs = []
1083 _set_last_step_outputs(ctx, last_step_tensor_outputs)
1612 def _set_last_step_outputs(ctx, last_step_tensor_outputs): argument
1617 ctx.last_step_outputs, last_step_tensor_outputs)
Done_device_strategy.py384 last_step_tensor_outputs = loop_result[1:]
386 ctx.last_step_outputs, last_step_tensor_outputs)
Dmirrored_strategy.py638 last_step_tensor_outputs = loop_result[1:]
640 ctx.last_step_outputs, last_step_tensor_outputs)