tf.contrib.distribute.Step

Class Step

Defined in tensorflow/contrib/distribute/python/step_fn.py.

Interface for performing each step of a training algorithm.

Properties

distribution

Methods

__init__

__init__(distribution)

__call__

__call__()

Perform one step of this training algorithm.

inputs

inputs()

For the generating the input to be passed to step().

step

step(inputs)

Perform the main computation of this training algorithm.