Home
last modified time | relevance | path

Searched defs:train_step (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/contrib/distribute/python/examples/
Dmnist_tf1_tpu.py106 def train_step(inputs): # pylint: disable=missing-docstring function
Dmnist_eager_multigpu.py113 def train_step(inputs): function
/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dlearning.py58 train_step = _slim.learning.train_step variable
/external/tensorflow/tensorflow/contrib/slim/python/slim/
Dlearning.py456 def train_step(sess, train_op, global_step, train_step_kwargs): function
/external/tensorflow/tensorflow/core/kernels/
Dsdca_ops.cc171 auto train_step = [&](const int64 begin, const int64 end) { in DoCompute() local
/external/tensorflow/tensorflow/contrib/compiler/
Dxla.py616 def train_step(): function
/external/tensorflow/tensorflow/python/tpu/
Dtpu_estimator.py1455 def train_step(loss): function