Home
last modified time | relevance | path

Searched refs:_set_last_step_outputs (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Done_device_strategy.py388 ctx._set_last_step_outputs(last_step_tensor_outputs_dict) # pylint: disable=protected-access
Dtpu_strategy.py1083 _set_last_step_outputs(ctx, last_step_tensor_outputs)
1612 def _set_last_step_outputs(ctx, last_step_tensor_outputs): function
1628 ctx._set_last_step_outputs(last_step_tensor_outputs_dict) # pylint: disable=protected-access
Dmirrored_strategy.py652 ctx._set_last_step_outputs(last_step_tensor_outputs_dict) # pylint: disable=protected-access
Dinput_lib.py2038 def _set_last_step_outputs(self, outputs): member in MultiStepContext