W3cubDocs

/TensorFlow Python

tf.contrib.distribute.StandardInputStep

Class StandardInputStep

Inherits From: Step

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

Step with a standard implementation of input handling.

Args:

  • input_dataset: a tf.data Dataset that provides input.

Properties

distribution

Methods

__init__

__init__(
    input_dataset,
    distribution
)

Initialize self. See help(type(self)) for accurate signature.

__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.

© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/contrib/distribute/StandardInputStep