Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/keras/distribute/
Dminimize_loss_test.py449 output_context.set_last_step_output(
459 output_context.set_last_step_output(
463 output_context.set_last_step_output(
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_distributed_v1.py113 ctx.set_last_step_output(label, output, reduce_op)
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib.py2044 def set_last_step_output(self, name, output, reduce_op=None): member in MultiStepContext