| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| D | learning.py | 58 train_step = _slim.learning.train_step variable
64 train_step_fn=train_step,
|
| /external/tensorflow/tensorflow/examples/tutorials/mnist/ |
| D | mnist_softmax_xla.py | 57 train_step = tf.train.GradientDescentOptimizer(0.5).minimize(cross_entropy)
77 sess.run(train_step,
86 sess.run(train_step, feed_dict={x: batch_xs, y_: batch_ys})
|
| D | mnist_with_summaries.py | 128 train_step = tf.train.AdamOptimizer(FLAGS.learning_rate).minimize(
168 summary, _ = sess.run([merged, train_step],
176 summary, _ = sess.run([merged, train_step], feed_dict=feed_dict(True))
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/tutorials/ |
| D | xla_compile.ipynb | 171 … `tf.Tensor` representing predicted probability of each target class, `train_step` is a `tf.Operat…
186 " train_step = tf.train.GradientDescentOptimizer(0.5).minimize(cross_entropy)\n",
188 " return y, train_step"
236 "y, train_step = build_mnist_model(images, labels)\n",
237 "with tf.control_dependencies([train_step]):\n",
290 …rains model. Evaluating `y` also triggers its control dependency node `train_step`, which updates …
|
| /external/tensorflow/tensorflow/python/debug/examples/ |
| D | debug_mnist.py | 112 train_step = tf.train.AdamOptimizer(FLAGS.learning_rate).minimize(
139 sess.run(train_step, feed_dict=feed_dict(True))
|
| /external/tensorflow/tensorflow/contrib/distribute/python/examples/ |
| D | mnist_eager_multigpu.py | 113 def train_step(inputs): function
136 dist_train = lambda it: strategy.experimental_run(train_step, it)
|
| D | mnist_tf1_tpu.py | 106 def train_step(inputs): # pylint: disable=missing-docstring function
133 strategy.experimental_run(train_step, train_iterator))
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | composable_model_test.py | 60 train_step = model.get_train_step(loss)
62 with ops.control_dependencies(train_step):
|
| /external/tensorflow/tensorflow/contrib/compiler/ |
| D | xla.py | 581 train_step, captured_scaffold_fn = self._make_train_step(
583 (loss,) = compile(train_step)
616 def train_step(): function
634 return train_step, captured_scaffold_fn
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sdca_ops.cc | 171 auto train_step = [&](const int64 begin, const int64 end) { in DoCompute() local
228 examples.num_examples(), kCostPerUnit, train_step); in DoCompute()
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/ |
| D | learning.py | 456 def train_step(sess, train_op, global_step, train_step_kwargs): function
533 train_step_fn=train_step,
|
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| D | train.py | 161 train_step = tf.train.GradientDescentOptimizer(
223 train_step,
|
| /external/tensorflow/tensorflow/core/profiler/g3doc/ |
| D | advise.md | 72 top 2 graph node: train_step/update_seq2seq/output_projection/w/ApplyAdam, cpu: 84.52ms, accelerato…
|
| /external/tensorflow/tensorflow/contrib/gan/python/ |
| D | train.py | 1241 cur_gen_loss, _ = slim_learning.train_step(
1248 cur_dis_loss, _ = slim_learning.train_step(
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| D | monitors_test.py | 255 train_step = gradient_descent.GradientDescentOptimizer(0.5).minimize(loss)
256 ops.get_default_session().run(train_step)
|
| /external/tensorflow/tensorflow/core/profiler/ |
| D | README.md | 178 top 2 graph node: train_step/update_seq2seq/output_projection/w/ApplyAdam, cpu: 84.52ms, accelerato…
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_estimator.py | 1455 def train_step(loss): function
1496 return (train_step, host_call, captured_scaffold_fn,
|