Home
last modified time | relevance | path

Searched refs:ModeKeys (Results 1 – 25 of 82) sorted by relevance

1234

/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dmodel_fn.py48 class ModeKeys(object): class
143 ModeKeys.validate(mode)
150 if mode == ModeKeys.TRAIN:
158 if mode in (ModeKeys.TRAIN, ModeKeys.EVAL):
170 if mode == ModeKeys.INFER or mode == ModeKeys.EVAL:
291 if self.mode == ModeKeys.TRAIN:
292 core_mode = core_model_fn_lib.ModeKeys.TRAIN
293 elif self.mode == ModeKeys.EVAL:
294 core_mode = core_model_fn_lib.ModeKeys.EVAL
295 elif self.mode == ModeKeys.INFER:
[all …]
Dhead_test.py123 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,
224 mode=model_fn.ModeKeys.EVAL,
240 mode=model_fn.ModeKeys.TRAIN,
256 mode=model_fn.ModeKeys.TRAIN,
274 mode=model_fn.ModeKeys.TRAIN,
[all …]
Dmodel_fn_test.py48 mode=model_fn.ModeKeys.INFER):
51 model_fn.ModeKeys.INFER,
84 predictions, None, mode=model_fn.ModeKeys.INFER)
94 predictions, output_alternatives, mode=model_fn.ModeKeys.INFER)
114 predictions, output_alternatives, mode=model_fn.ModeKeys.INFER)
131 predictions, output_alternatives, mode=model_fn.ModeKeys.INFER)
153 predictions, output_alternatives, mode=model_fn.ModeKeys.INFER)
176 predictions, output_alternatives, mode=model_fn.ModeKeys.INFER)
197 predictions, output_alternatives, mode=model_fn.ModeKeys.INFER)
219 predictions, output_alternatives, mode=model_fn.ModeKeys.INFER)
[all …]
Destimator_input_test.py111 assert mode in (model_fn.ModeKeys.TRAIN, model_fn.ModeKeys.EVAL,
112 model_fn.ModeKeys.INFER)
125 assert mode in (model_fn.ModeKeys.TRAIN, model_fn.ModeKeys.EVAL,
126 model_fn.ModeKeys.INFER)
140 assert mode in (model_fn.ModeKeys.TRAIN, model_fn.ModeKeys.EVAL,
141 model_fn.ModeKeys.INFER)
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_generator.py36 from tensorflow.python.keras.utils.mode_keys import ModeKeys
56 mode=ModeKeys.TRAIN,
180 if mode == ModeKeys.PREDICT:
188 backend.set_eager_learning_phase(1 if mode == ModeKeys.TRAIN else 0)
200 if mode == ModeKeys.TRAIN:
232 if mode == ModeKeys.TRAIN:
314 mode=ModeKeys.TEST,
322 if mode == ModeKeys.TRAIN:
339 if mode == ModeKeys.TRAIN:
345 fit_generator = functools.partial(model_iteration, mode=ModeKeys.TRAIN)
[all …]
Dtraining_arrays.py36 from tensorflow.python.keras.utils.mode_keys import ModeKeys
61 mode=ModeKeys.TRAIN,
140 if mode == ModeKeys.TRAIN:
147 learning_phase=(1 if mode == ModeKeys.TRAIN else 0))
191 model, val_iterator, val_targets, val_sample_weights, ModeKeys.TEST)
219 if mode == ModeKeys.PREDICT:
240 if mode == ModeKeys.TRAIN:
282 if mode == ModeKeys.TRAIN:
384 model, ModeKeys.TRAIN)
395 mode=ModeKeys.TEST,
[all …]
Ddistributed_training_utils.py37 from tensorflow.python.keras.utils.mode_keys import ModeKeys
441 allow_partial_batch = (mode == ModeKeys.TRAIN or
442 (mode == ModeKeys.PREDICT
575 if mode == ModeKeys.PREDICT:
583 if mode == ModeKeys.TRAIN and not isinstance(K.symbolic_learning_phase(),
655 if mode == ModeKeys.PREDICT and inputs is not None: # TPU predict case
707 if mode == ModeKeys.PREDICT and inputs is not None: # TPU predict case
728 if mode == ModeKeys.TRAIN:
796 with_loss_tensor=(mode != ModeKeys.PREDICT))
836 with_loss_tensor=(mode != ModeKeys.PREDICT))
[all …]
Dtraining_distributed.py37 from tensorflow.python.keras.utils.mode_keys import ModeKeys
72 batch_size, mode=ModeKeys.TRAIN))
181 batch_size, mode=ModeKeys.PREDICT)
220 if mode == ModeKeys.PREDICT:
248 if mode == ModeKeys.PREDICT:
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)
476 mode = ModeKeys.TEST
[all …]
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
Dhead_test.py62 for mode in [estimator_lib.ModeKeys.TRAIN, estimator_lib.ModeKeys.EVAL,
63 estimator_lib.ModeKeys.PREDICT]:
120 features=features, labels=None, mode=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]:
288 mode=estimator_lib.ModeKeys.PREDICT)
[all …]
Dhead.py97 mode = estimator_lib.ModeKeys.TRAIN
114 mode = estimator_lib.ModeKeys.EVAL
143 predictions=prediction, mode=estimator_lib.ModeKeys.PREDICT)
151 features, estimator_lib.ModeKeys.EVAL)
159 features=no_state_features, mode=estimator_lib.ModeKeys.EVAL)
161 mode=estimator_lib.ModeKeys.PREDICT,
248 features, update_statistics=(mode == estimator_lib.ModeKeys.TRAIN))
256 if (mode == estimator_lib.ModeKeys.TRAIN or
257 mode == estimator_lib.ModeKeys.EVAL):
259 elif mode == estimator_lib.ModeKeys.PREDICT:
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Destimator_utils.py29 model_fn_lib.ModeKeys.TRAIN: contrib_model_fn_lib.ModeKeys.TRAIN,
30 model_fn_lib.ModeKeys.EVAL: contrib_model_fn_lib.ModeKeys.EVAL,
31 model_fn_lib.ModeKeys.PREDICT: contrib_model_fn_lib.ModeKeys.INFER
/external/tensorflow/tensorflow/contrib/gan/python/estimator/python/
Dstargan_estimator_impl.py161 model_fn_lib.ModeKeys.TRAIN, model_fn_lib.ModeKeys.EVAL,
162 model_fn_lib.ModeKeys.PREDICT
166 if mode == model_fn_lib.ModeKeys.PREDICT:
196 if mode == model_fn_lib.ModeKeys.PREDICT:
215 if mode == model_fn_lib.ModeKeys.PREDICT:
220 if mode == model_fn_lib.ModeKeys.EVAL:
268 generator_fn, mode=model_fn_lib.ModeKeys.PREDICT)
321 mode=model_fn_lib.ModeKeys.EVAL,
340 mode=model_fn_lib.ModeKeys.TRAIN,
Dgan_estimator_impl.py177 if mode not in [model_fn_lib.ModeKeys.TRAIN, model_fn_lib.ModeKeys.EVAL,
178 model_fn_lib.ModeKeys.PREDICT]:
204 if mode == model_fn_lib.ModeKeys.PREDICT:
223 if mode == model_fn_lib.ModeKeys.PREDICT:
232 if mode == model_fn_lib.ModeKeys.EVAL:
278 mode=model_fn_lib.ModeKeys.PREDICT)
320 mode=model_fn_lib.ModeKeys.EVAL,
336 mode=model_fn_lib.ModeKeys.TRAIN,
Dtpu_gan_estimator_impl.py226 if mode not in [model_fn_lib.ModeKeys.TRAIN, model_fn_lib.ModeKeys.EVAL,
227 model_fn_lib.ModeKeys.PREDICT]:
265 if mode == model_fn_lib.ModeKeys.TRAIN:
267 elif mode == model_fn_lib.ModeKeys.EVAL:
278 if mode == model_fn_lib.ModeKeys.PREDICT:
281 elif mode == model_fn_lib.ModeKeys.EVAL:
358 mode=model_fn_lib.ModeKeys.EVAL,
398 mode=model_fn_lib.ModeKeys.TRAIN,
Dstargan_estimator_test.py64 @parameterized.named_parameters(('train', model_fn_lib.ModeKeys.TRAIN),
65 ('eval', model_fn_lib.ModeKeys.EVAL),
66 ('predict', model_fn_lib.ModeKeys.PREDICT))
85 if mode == model_fn_lib.ModeKeys.PREDICT:
169 @parameterized.named_parameters(('train', model_fn_lib.ModeKeys.TRAIN),
170 ('eval', model_fn_lib.ModeKeys.EVAL),
171 ('predict', model_fn_lib.ModeKeys.PREDICT))
184 if mode == model_fn_lib.ModeKeys.PREDICT:
186 elif mode == model_fn_lib.ModeKeys.TRAIN:
190 elif mode == model_fn_lib.ModeKeys.EVAL:
Dgan_estimator_test.py71 ('train', model_fn_lib.ModeKeys.TRAIN),
72 ('eval', model_fn_lib.ModeKeys.EVAL),
73 ('predict', model_fn_lib.ModeKeys.PREDICT))
77 is_predict = mode == model_fn_lib.ModeKeys.PREDICT
87 if mode == model_fn_lib.ModeKeys.PREDICT:
146 ('train', model_fn_lib.ModeKeys.TRAIN),
147 ('eval', model_fn_lib.ModeKeys.EVAL),
148 ('predict', model_fn_lib.ModeKeys.PREDICT))
162 if mode == model_fn_lib.ModeKeys.PREDICT:
164 elif mode == model_fn_lib.ModeKeys.TRAIN:
[all …]
Dhead_impl.py202 if mode == model_fn_lib.ModeKeys.PREDICT:
204 mode=model_fn_lib.ModeKeys.PREDICT,
209 elif mode == model_fn_lib.ModeKeys.EVAL:
229 mode=model_fn_lib.ModeKeys.EVAL,
233 elif mode == model_fn_lib.ModeKeys.TRAIN:
243 mode=model_fn_lib.ModeKeys.TRAIN,
Dtpu_gan_estimator_test.py107 ('joint_train', model_fn_lib.ModeKeys.TRAIN, True),
108 ('train_sequential', model_fn_lib.ModeKeys.TRAIN, False),
109 ('eval', model_fn_lib.ModeKeys.EVAL, None),
110 ('predict', model_fn_lib.ModeKeys.PREDICT, None))
128 if mode == model_fn_lib.ModeKeys.PREDICT:
131 elif mode == model_fn_lib.ModeKeys.TRAIN:
135 elif mode == model_fn_lib.ModeKeys.EVAL:
/external/tensorflow/tensorflow/python/saved_model/model_utils/
Dexport_utils.py34 from tensorflow.python.saved_model.model_utils.mode_keys import KerasModeKeys as ModeKeys unknown
40 ModeKeys.PREDICT: [tag_constants.SERVING],
41 ModeKeys.TRAIN: [tag_constants.TRAINING],
42 ModeKeys.TEST: [tag_constants.EVAL]})
50 ModeKeys.PREDICT: signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY,
51 ModeKeys.TRAIN: signature_constants.DEFAULT_TRAIN_SIGNATURE_DEF_KEY,
52 ModeKeys.TEST: signature_constants.DEFAULT_EVAL_SIGNATURE_DEF_KEY})
/external/tensorflow/tensorflow/python/keras/saving/
Dsaved_model.py188 _export_mode(mode_keys.ModeKeys.TRAIN, has_saved_vars, **export_args)
190 _export_mode(mode_keys.ModeKeys.TEST, has_saved_vars, **export_args)
197 _export_mode(mode_keys.ModeKeys.PREDICT, has_saved_vars, **export_args)
234 compile_clone = (mode != mode_keys.ModeKeys.PREDICT)
241 mode == mode_keys.ModeKeys.TRAIN):
263 if mode == mode_keys.ModeKeys.TRAIN:
266 elif mode == mode_keys.ModeKeys.TEST:
340 serving_only=(mode == mode_keys.ModeKeys.PREDICT))
/external/tensorflow/tensorflow/python/tpu/
Dtpu_context.py254 if mode != model_fn_lib.ModeKeys.EVAL else config.evaluation_master)
397 return (mode == model_fn_lib.ModeKeys.TRAIN and
411 (mode != model_fn_lib.ModeKeys.TRAIN and
447 if mode == model_fn_lib.ModeKeys.EVAL and not self._eval_on_tpu:
459 if mode == model_fn_lib.ModeKeys.TRAIN:
461 elif mode == model_fn_lib.ModeKeys.EVAL:
463 elif mode == model_fn_lib.ModeKeys.PREDICT:
515 if mode == model_fn_lib.ModeKeys.EVAL else run_config.master)
634 if mode == model_fn_lib.ModeKeys.TRAIN:
641 elif mode == model_fn_lib.ModeKeys.EVAL:
[all …]
D_tpu_estimator_embedding.py233 if mode == model_fn_lib.ModeKeys.TRAIN:
238 if mode == model_fn_lib.ModeKeys.TRAIN:
242 elif (mode == model_fn_lib.ModeKeys.EVAL or
243 mode == model_fn_lib.ModeKeys.PREDICT):
256 if mode == model_fn_lib.ModeKeys.EVAL else self._run_config.master)
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dwals_test.py161 if mode == model_fn.ModeKeys.INFER or mode == model_fn.ModeKeys.EVAL:
168 if mode == model_fn.ModeKeys.INFER and projection_weights is not None:
262 mode=model_fn.ModeKeys.TRAIN,
273 mode=model_fn.ModeKeys.TRAIN,
284 mode=model_fn.ModeKeys.TRAIN,
291 mode=model_fn.ModeKeys.INFER,
308 mode=model_fn.ModeKeys.INFER,
327 mode=model_fn.ModeKeys.TRAIN,
333 mode=model_fn.ModeKeys.EVAL,
351 mode=model_fn.ModeKeys.EVAL,
Dwals.py160 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,
318 elif mode == model_fn.ModeKeys.INFER:
342 mode=model_fn.ModeKeys.INFER,
350 elif mode == model_fn.ModeKeys.EVAL:
364 mode=model_fn.ModeKeys.EVAL,
/external/tensorflow/tensorflow/python/keras/
Dcallbacks.py39 from tensorflow.python.keras.utils.mode_keys import ModeKeys
60 mode=ModeKeys.TRAIN):
87 if mode == ModeKeys.TRAIN:
121 mode=ModeKeys.TRAIN):
144 if mode != ModeKeys.PREDICT and hasattr(model, 'metrics_names'):
168 if mode in {ModeKeys.TRAIN, ModeKeys.TEST}:
240 if mode == ModeKeys.TRAIN:
242 elif mode == ModeKeys.TEST:
249 if mode == ModeKeys.TRAIN:
251 elif mode == ModeKeys.TEST:
[all …]

1234