Home
last modified time | relevance | path

Searched refs:set_last_step_output (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/distribute/python/
Dminimize_loss_test.py394 output_context.set_last_step_output(
404 output_context.set_last_step_output(
408 output_context.set_last_step_output(
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_distributed.py249 ctx.set_last_step_output(label, output)
257 ctx.set_last_step_output(label, output, reduce_op)
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib.py599 def set_last_step_output(self, name, output, reduce_op=None): member in MultiStepContext