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/
Dtpu_strategy.py380 _set_last_step_outputs(ctx, last_step_tensor_outputs)
655 def _set_last_step_outputs(ctx, last_step_tensor_outputs): function
672 ctx._set_last_step_outputs(last_step_tensor_outputs_dict) # pylint: disable=protected-access
Done_device_strategy.py138 ctx._set_last_step_outputs(last_step_tensor_outputs_dict) # pylint: disable=protected-access
Dinput_lib.py593 def _set_last_step_outputs(self, outputs): member in MultiStepContext
Dmirrored_strategy.py645 ctx._set_last_step_outputs(last_step_tensor_outputs_dict) # pylint: disable=protected-access