| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | head_test.py | 123 mode=model_fn.ModeKeys.TRAIN,
149 mode=model_fn.ModeKeys.TRAIN,
167 mode=model_fn.ModeKeys.TRAIN,
183 mode=model_fn.ModeKeys.TRAIN,
193 mode=model_fn.ModeKeys.TRAIN,
213 mode=model_fn.ModeKeys.TRAIN,
240 mode=model_fn.ModeKeys.TRAIN,
256 mode=model_fn.ModeKeys.TRAIN,
274 mode=model_fn.ModeKeys.TRAIN,
291 mode=model_fn.ModeKeys.TRAIN,
[all …]
|
| D | model_fn.py | 60 TRAIN = 'train' variable in ModeKeys
66 if key not in (cls.TRAIN, cls.EVAL, cls.INFER):
150 if mode == ModeKeys.TRAIN:
158 if mode in (ModeKeys.TRAIN, ModeKeys.EVAL):
291 if self.mode == ModeKeys.TRAIN:
292 core_mode = core_model_fn_lib.ModeKeys.TRAIN
|
| /external/tensorflow/tensorflow/python/saved_model/model_utils/ |
| D | mode_keys.py | 35 TRAIN = 'train' variable in KerasModeKeys
51 TRAIN = 'train' variable in EstimatorModeKeys
65 return mode in [KerasModeKeys.TRAIN, EstimatorModeKeys.TRAIN]
95 return KerasModeKeys.TRAIN
|
| D | mode_keys_test.py | 39 _ = mode_map[mode_keys.KerasModeKeys.TRAIN]
41 _ = mode_map[mode_keys.EstimatorModeKeys.TRAIN]
|
| D | export_utils.py | 41 ModeKeys.TRAIN: [tag_constants.TRAINING],
51 ModeKeys.TRAIN: signature_constants.DEFAULT_TRAIN_SIGNATURE_DEF_KEY,
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_arrays.py | 61 mode=ModeKeys.TRAIN,
140 if mode == ModeKeys.TRAIN:
147 learning_phase=(1 if mode == ModeKeys.TRAIN else 0))
240 if mode == ModeKeys.TRAIN:
282 if mode == ModeKeys.TRAIN:
384 model, ModeKeys.TRAIN)
406 if mode == ModeKeys.TRAIN:
423 if mode == ModeKeys.TRAIN:
496 if mode == ModeKeys.TRAIN and not isinstance(K.symbolic_learning_phase(),
525 fit_loop = functools.partial(model_iteration, mode=ModeKeys.TRAIN)
|
| D | training_generator.py | 56 mode=ModeKeys.TRAIN,
188 backend.set_eager_learning_phase(1 if mode == ModeKeys.TRAIN else 0)
200 if mode == ModeKeys.TRAIN:
232 if mode == ModeKeys.TRAIN:
322 if mode == ModeKeys.TRAIN:
339 if mode == ModeKeys.TRAIN:
345 fit_generator = functools.partial(model_iteration, mode=ModeKeys.TRAIN)
415 arg_name = 'steps_per_epoch' if mode == ModeKeys.TRAIN else 'steps'
523 if mode == ModeKeys.TRAIN:
|
| D | distributed_training_utils.py | 441 allow_partial_batch = (mode == ModeKeys.TRAIN or
583 if mode == ModeKeys.TRAIN and not isinstance(K.symbolic_learning_phase(),
728 if mode == ModeKeys.TRAIN:
860 if model._distribution_strategy and mode == ModeKeys.TRAIN:
881 for mode in [ModeKeys.TRAIN, ModeKeys.TEST, ModeKeys.PREDICT]:
|
| D | training_distributed.py | 72 batch_size, mode=ModeKeys.TRAIN))
308 mode = ModeKeys.TRAIN
326 step_fn = _make_step_fn(model, ModeKeys.TRAIN, current_strategy, out_labels)
427 model, ModeKeys.TRAIN)
449 model, ModeKeys.TRAIN)
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/state_space_models/ |
| D | varma_test.py | 47 mode=estimator_lib.ModeKeys.TRAIN)
66 mode=estimator_lib.ModeKeys.TRAIN)
86 mode=estimator_lib.ModeKeys.TRAIN)
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| D | head_test.py | 62 for mode in [estimator_lib.ModeKeys.TRAIN, estimator_lib.ModeKeys.EVAL,
201 for mode in [estimator_lib.ModeKeys.TRAIN, estimator_lib.ModeKeys.EVAL]:
211 for mode in [estimator_lib.ModeKeys.TRAIN, estimator_lib.ModeKeys.EVAL]:
221 for mode in [estimator_lib.ModeKeys.TRAIN, estimator_lib.ModeKeys.EVAL]:
235 for mode in [estimator_lib.ModeKeys.TRAIN, estimator_lib.ModeKeys.EVAL]:
249 for mode in [estimator_lib.ModeKeys.TRAIN, estimator_lib.ModeKeys.EVAL]:
263 for mode in [estimator_lib.ModeKeys.TRAIN, estimator_lib.ModeKeys.EVAL]:
|
| D | state_management_test.py | 129 model=stub_model, features=features, mode=estimator_lib.ModeKeys.TRAIN)
177 mode=estimator_lib.ModeKeys.TRAIN)
180 mode=estimator_lib.ModeKeys.TRAIN)
223 model=stub_model, features=features, mode=estimator_lib.ModeKeys.TRAIN)
|
| D | head.py | 97 mode = estimator_lib.ModeKeys.TRAIN
248 features, update_statistics=(mode == estimator_lib.ModeKeys.TRAIN))
256 if (mode == estimator_lib.ModeKeys.TRAIN or
267 if mode == estimator_lib.ModeKeys.TRAIN:
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| D | estimator_utils.py | 29 model_fn_lib.ModeKeys.TRAIN: contrib_model_fn_lib.ModeKeys.TRAIN,
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | wals.py | 160 and mode == model_fn.ModeKeys.TRAIN)
162 mode == model_fn.ModeKeys.TRAIN)
185 if mode == model_fn.ModeKeys.TRAIN:
310 mode=model_fn.ModeKeys.TRAIN,
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | callbacks.py | 60 mode=ModeKeys.TRAIN):
87 if mode == ModeKeys.TRAIN:
121 mode=ModeKeys.TRAIN):
168 if mode in {ModeKeys.TRAIN, ModeKeys.TEST}:
240 if mode == ModeKeys.TRAIN:
249 if mode == ModeKeys.TRAIN:
257 self._call_batch_hook(ModeKeys.TRAIN, 'begin', batch, logs=logs)
260 self._call_batch_hook(ModeKeys.TRAIN, 'end', batch, logs=logs)
300 self._call_batch_hook(ModeKeys.TRAIN, 'begin', batch, logs=logs)
309 self._call_batch_hook(ModeKeys.TRAIN, 'end', batch, logs=logs)
|
| /external/tensorflow/tensorflow/contrib/gan/python/estimator/python/ |
| D | tpu_gan_estimator_impl.py | 226 if mode not in [model_fn_lib.ModeKeys.TRAIN, model_fn_lib.ModeKeys.EVAL,
265 if mode == model_fn_lib.ModeKeys.TRAIN:
398 mode=model_fn_lib.ModeKeys.TRAIN,
|
| D | stargan_estimator_test.py | 64 @parameterized.named_parameters(('train', model_fn_lib.ModeKeys.TRAIN),
169 @parameterized.named_parameters(('train', model_fn_lib.ModeKeys.TRAIN),
186 elif mode == model_fn_lib.ModeKeys.TRAIN:
|
| D | gan_estimator_test.py | 71 ('train', model_fn_lib.ModeKeys.TRAIN),
146 ('train', model_fn_lib.ModeKeys.TRAIN),
164 elif mode == model_fn_lib.ModeKeys.TRAIN:
187 model_fn_lib.ModeKeys.TRAIN,
|
| D | head_impl.py | 233 elif mode == model_fn_lib.ModeKeys.TRAIN:
243 mode=model_fn_lib.ModeKeys.TRAIN,
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | saved_model.py | 188 _export_mode(mode_keys.ModeKeys.TRAIN, has_saved_vars, **export_args)
241 mode == mode_keys.ModeKeys.TRAIN):
263 if mode == mode_keys.ModeKeys.TRAIN:
|
| /external/tensorflow/tensorflow/examples/tutorials/layers/ |
| D | cnn_mnist.py | 82 inputs=dense, rate=0.4, training=mode == tf.estimator.ModeKeys.TRAIN)
103 if mode == tf.estimator.ModeKeys.TRAIN:
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_context.py | 397 return (mode == model_fn_lib.ModeKeys.TRAIN and
411 (mode != model_fn_lib.ModeKeys.TRAIN and
459 if mode == model_fn_lib.ModeKeys.TRAIN:
634 if mode == model_fn_lib.ModeKeys.TRAIN:
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.estimator.-mode-keys.pbtxt | 14 name: "TRAIN"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.estimator.-mode-keys.pbtxt | 14 name: "TRAIN"
|