Searched refs:_set_last_step_outputs (Results 1 – 4 of 4) sorted by relevance
380 _set_last_step_outputs(ctx, last_step_tensor_outputs)655 def _set_last_step_outputs(ctx, last_step_tensor_outputs): function672 ctx._set_last_step_outputs(last_step_tensor_outputs_dict) # pylint: disable=protected-access
138 ctx._set_last_step_outputs(last_step_tensor_outputs_dict) # pylint: disable=protected-access
593 def _set_last_step_outputs(self, outputs): member in MultiStepContext
645 ctx._set_last_step_outputs(last_step_tensor_outputs_dict) # pylint: disable=protected-access