StandardInputStep
Inherits From: Step
Defined in tensorflow/contrib/distribute/python/step_fn.py
.
Step with a standard implementation of input handling.
input_dataset
: a tf.data Dataset that provides input.distribution
__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