Home
last modified time | relevance | path

Searched refs:train_op_fn (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dhead_test.py124 train_op_fn=head_lib.no_op_train_fn,
150 train_op_fn=head_lib.no_op_train_fn,
168 train_op_fn=head_lib.no_op_train_fn,
184 train_op_fn=head_lib.no_op_train_fn,
194 train_op_fn=head_lib.no_op_train_fn,
214 train_op_fn=head_lib.no_op_train_fn,
225 train_op_fn=head_lib.no_op_train_fn,
241 train_op_fn=head_lib.no_op_train_fn,
257 train_op_fn=head_lib.no_op_train_fn,
275 train_op_fn=head_lib.no_op_train_fn,
[all …]
Dhead.py153 train_op_fn=None, argument
637 train_op_fn=None, argument
665 if train_op_fn is None:
668 train_op = _train_op(loss, labels, train_op_fn, centered_bias,
733 train_op_fn=None, argument
753 train_op_fn=train_op_fn,
854 train_op_fn=None, argument
875 train_op_fn=train_op_fn,
1059 train_op_fn=None, argument
1080 train_op_fn=train_op_fn,
[all …]
Ddebug.py110 def train_op_fn(loss): function
118 train_op_fn=train_op_fn,
Dcomposable_model_test.py70 train_op_fn=_train_op_fn,
Dlinear.py190 train_op_fn=_train_op_fn,
275 train_op_fn=_train_op_fn,
Ddnn.py213 train_op_fn=_train_op_fn,
Ddnn_linear_combined.py373 train_op_fn=_make_training_op,
Ddnn_linear_combined_test.py78 self, mode, features, labels=None, train_op_fn=None, logits=None, argument
87 train_op=train_op_fn(loss),
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Dmodel.py162 train_op_fn=_train_op_fn,
171 train_op_fn=_train_op_fn,
186 train_op_fn=_train_op_fn,
381 train_op_fn=_train_op_fn,
390 train_op_fn=_train_op_fn,
406 train_op_fn=_train_op_fn,
Ddnn_tree_combined_estimator.py329 train_op_fn=_no_train_op_fn,
337 train_op_fn=dnn_train_op_fn,
343 train_op_fn=_tree_train_op_fn,
361 train_op_fn=_no_train_op_fn,
369 train_op_fn=dnn_train_op_fn,
375 train_op_fn=_tree_train_op_fn,
/external/tensorflow/tensorflow/contrib/gan/python/estimator/python/
Dhead_impl.py174 train_op_fn=tfgan_train.gan_train_ops): argument
234 if train_op_fn is None:
238 train_ops = train_op_fn(gan_model, gan_loss, self._generator_optimizer,
Dgan_estimator_impl.py328 get_hooks_fn, train_op_fn=tfgan_train.gan_train_ops, is_chief=True): argument
331 train_ops = train_op_fn(gan_model, gan_loss, generator_optimizer,
Dstargan_estimator_impl.py332 train_op_fn=tfgan_train.gan_train_ops): argument
335 train_ops = train_op_fn(gan_model, gan_loss, generator_optimizer,
/external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/
Destimator_test.py89 train_op_fn=head_lib.no_op_train_fn,
/external/tensorflow/tensorflow/contrib/tensor_forest/client/
Drandom_forest.py262 train_op_fn=_train_fn,
299 train_op_fn=_train_fn,
/external/tensorflow/tensorflow/contrib/linear_optimizer/python/
Dsdca_estimator.py172 train_op_fn=_train_op_fn,