Searched refs:_make_step_fn (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_distributed.py | 200 def _make_step_fn(model, mode, strategy, output_labels): function
326 step_fn = _make_step_fn(model, ModeKeys.TRAIN, current_strategy, out_labels)
488 step_fn = _make_step_fn(model, ModeKeys.TEST, current_strategy, out_labels)
629 step_fn = _make_step_fn(model, ModeKeys.PREDICT, current_strategy, out_labels)
|