Searched defs:TRAIN (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/estimator/
H A Dmodel_fn.py43 * `TRAIN`: training mode.
48 TRAIN = 'train' variable in class:ModeKeys
85 * For `mode == ModeKeys.TRAIN`: required fields are `loss` and `train_op`.
110 if (mode == tf.estimator.ModeKeys.TRAIN or
115 if mode == tf.estimator.ModeKeys.TRAIN:
173 if mode == ModeKeys.TRAIN:
180 if mode in (ModeKeys.TRAIN, ModeKeys.EVAL):
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
H A Dmodel_fn.py47 * `TRAIN`: training mode.
52 TRAIN = 'train' variable in class:ModeKeys
58 if key not in (cls.TRAIN, cls.EVAL, cls.INFER):
134 if mode == ModeKeys.TRAIN:
142 if mode in (ModeKeys.TRAIN, ModeKeys.EVAL):
275 if self.mode == ModeKeys.TRAIN:
276 core_mode = core_model_fn_lib.ModeKeys.TRAIN

Completed in 424 milliseconds