Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/gan/python/estimator/python/
Dstargan_estimator_impl.py29 from tensorflow.python.estimator import model_fn as model_fn_lib unknown
161 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:
216 estimator_spec = model_fn_lib.EstimatorSpec(
220 if mode == model_fn_lib.ModeKeys.EVAL:
268 generator_fn, mode=model_fn_lib.ModeKeys.PREDICT)
320 return model_fn_lib.EstimatorSpec(
[all …]
Dgan_estimator_impl.py29 from tensorflow.python.estimator import model_fn as model_fn_lib unknown
177 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:
224 estimator_spec = model_fn_lib.EstimatorSpec(
232 if mode == model_fn_lib.ModeKeys.EVAL:
278 mode=model_fn_lib.ModeKeys.PREDICT)
319 return model_fn_lib.EstimatorSpec(
320 mode=model_fn_lib.ModeKeys.EVAL,
[all …]
Dhead_impl.py25 from tensorflow.python.estimator import model_fn as model_fn_lib unknown
202 if mode == model_fn_lib.ModeKeys.PREDICT:
203 return model_fn_lib.EstimatorSpec(
204 mode=model_fn_lib.ModeKeys.PREDICT,
209 elif mode == model_fn_lib.ModeKeys.EVAL:
228 return model_fn_lib.EstimatorSpec(
229 mode=model_fn_lib.ModeKeys.EVAL,
233 elif mode == model_fn_lib.ModeKeys.TRAIN:
241 return model_fn_lib.EstimatorSpec(
243 mode=model_fn_lib.ModeKeys.TRAIN,
Dtpu_gan_estimator_impl.py27 from tensorflow.python.estimator import model_fn as model_fn_lib unknown
226 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.py30 from tensorflow.python.estimator import model_fn as model_fn_lib unknown
64 @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:
[all …]
Dgan_estimator_test.py34 from tensorflow.python.estimator import model_fn as model_fn_lib unknown
71 ('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:
[all …]
Dhead_test.py24 from tensorflow.python.estimator import model_fn as model_fn_lib unknown
83 spec = self._test_modes_helper(model_fn_lib.ModeKeys.PREDICT)
88 self._test_modes_helper(model_fn_lib.ModeKeys.EVAL)
91 self._test_modes_helper(model_fn_lib.ModeKeys.TRAIN)
Dtpu_gan_estimator_test.py35 from tensorflow.python.estimator import model_fn as model_fn_lib unknown
107 ('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:
Dlatent_gan_estimator_impl.py76 from tensorflow.python.estimator import model_fn as model_fn_lib unknown
131 return model_fn_lib.EstimatorSpec(mode=mode,
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Destimator_utils.py25 from tensorflow.python.estimator import model_fn as model_fn_lib unknown
29 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/python/tpu/
Dtpu_context.py24 from tensorflow.python.estimator import model_fn as model_fn_lib unknown
254 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:
[all …]
D_tpu_estimator_embedding.py23 from tensorflow.python.estimator import model_fn as model_fn_lib unknown
233 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)
Dtpu_estimator.py42 from tensorflow.python.estimator import model_fn as model_fn_lib unknown
266 class TPUEstimatorSpec(model_fn_lib._TPUEstimatorSpec): # pylint: disable=protected-access
367 return model_fn_lib.EstimatorSpec(
787 if ctx.mode == model_fn_lib.ModeKeys.PREDICT:
865 if ctx.mode == model_fn_lib.ModeKeys.PREDICT:
958 if ctx.mode == model_fn_lib.ModeKeys.PREDICT:
1279 if self._ctx.mode != model_fn_lib.ModeKeys.PREDICT else
1319 if self._ctx.mode != model_fn_lib.ModeKeys.PREDICT else
1466 … if isinstance(estimator_spec, model_fn_lib._TPUEstimatorSpec): # pylint: disable=protected-access
1489 … if (isinstance(estimator_spec, model_fn_lib._TPUEstimatorSpec) # pylint: disable=protected-access
[all …]
/external/tensorflow/tensorflow/contrib/tensor_forest/client/
Drandom_forest.py25 from tensorflow.contrib.learn.python.learn.estimators import model_fn as model_fn_lib unknown
198 if (mode == model_fn_lib.ModeKeys.INFER or
199 (local_eval and mode == model_fn_lib.ModeKeys.EVAL)):
221 if labels is not None and mode == model_fn_lib.ModeKeys.TRAIN:
511 if (mode == model_fn_lib.ModeKeys.EVAL or
512 mode == model_fn_lib.ModeKeys.INFER):
521 if (mode == model_fn_lib.ModeKeys.EVAL or
522 mode == model_fn_lib.ModeKeys.TRAIN):
528 if mode == model_fn_lib.ModeKeys.TRAIN:
531 return model_fn_lib.ModelFnOps(
/external/tensorflow/tensorflow/contrib/compiler/
Dxla.py29 from tensorflow.python.estimator import model_fn as model_fn_lib unknown
580 if mode == model_fn_lib.ModeKeys.TRAIN:
584 return model_fn_lib.EstimatorSpec(
589 elif mode == model_fn_lib.ModeKeys.EVAL:
603 return model_fn_lib.EstimatorSpec(
619 model_fn_lib.ModeKeys.TRAIN, params)
644 model_fn_lib.ModeKeys.EVAL, params)
711 if estimator_spec.mode == model_fn_lib.ModeKeys.EVAL:
Dxla_test.py30 from tensorflow.python.estimator import model_fn as model_fn_lib unknown
44 _TRAIN = model_fn_lib.ModeKeys.TRAIN
45 _EVAL = model_fn_lib.ModeKeys.EVAL
275 return model_fn_lib.EstimatorSpec(
359 return model_fn_lib.EstimatorSpec(
405 return model_fn_lib.EstimatorSpec(
422 return model_fn_lib.EstimatorSpec(
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dlogistic_regressor.py33 from tensorflow.contrib.learn.python.learn.estimators import model_fn as model_fn_lib unknown
55 if mode == model_fn_lib.ModeKeys.EVAL:
62 return model_fn_lib.ModelFnOps(
Dstate_saving_rnn_estimator_test.py29 from tensorflow.contrib.learn.python.learn.estimators import model_fn as model_fn_lib unknown
244 mode = model_fn_lib.ModeKeys.TRAIN
356 model_fn_ops = self._getModelFnOpsForMode(model_fn_lib.ModeKeys.TRAIN)
364 model_fn_ops = self._getModelFnOpsForMode(model_fn_lib.ModeKeys.EVAL)
372 model_fn_ops = self._getModelFnOpsForMode(model_fn_lib.ModeKeys.INFER)
404 if mode == model_fn_lib.ModeKeys.INFER:
429 model_fn_lib.ModeKeys.TRAIN, seed=1234),
439 model_fn_lib.ModeKeys.INFER, seed=4321),
Dcomposable_model_test.py27 from tensorflow.contrib.learn.python.learn.estimators import model_fn as model_fn_lib unknown
50 if mode == model_fn_lib.ModeKeys.TRAIN:
52 elif mode == model_fn_lib.ModeKeys.EVAL:
Ddynamic_rnn_estimator_test.py30 from tensorflow.contrib.learn.python.learn.estimators import model_fn as model_fn_lib unknown
152 mode != model_fn_lib.ModeKeys.INFER else None)
225 model_fn_ops = self._GetModelFnOpsForMode(model_fn_lib.ModeKeys.TRAIN)
233 model_fn_ops = self._GetModelFnOpsForMode(model_fn_lib.ModeKeys.EVAL)
241 model_fn_ops = self._GetModelFnOpsForMode(model_fn_lib.ModeKeys.INFER)
273 if mode == model_fn_lib.ModeKeys.INFER:
296 estimator_fn().fit(input_fn=get_input_fn(model_fn_lib.ModeKeys.TRAIN),
305 input_fn=get_input_fn(model_fn_lib.ModeKeys.INFER),
Destimator.py48 from tensorflow.contrib.learn.python.learn.estimators import model_fn as model_fn_lib unknown
1229 if isinstance(model_fn_results, model_fn_lib.ModelFnOps):
1236 model_fn_ops = model_fn_lib.ModelFnOps(
1264 return self._call_model_fn(features, labels, model_fn_lib.ModeKeys.TRAIN)
1292 model_fn_lib.ModeKeys.EVAL, metrics)
1314 return self._call_model_fn(features, labels, model_fn_lib.ModeKeys.INFER)
1399 model_fn_lib.ModeKeys.INFER)
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
Dkeras_support.py71 from tensorflow.python.estimator import model_fn as model_fn_lib unknown
856 if mode not in [model_fn_lib.ModeKeys.TRAIN, model_fn_lib.ModeKeys.EVAL]:
883 if mode not in [model_fn_lib.ModeKeys.TRAIN, model_fn_lib.ModeKeys.EVAL]:
901 if mode not in [model_fn_lib.ModeKeys.TRAIN, model_fn_lib.ModeKeys.EVAL]:
943 K.set_learning_phase(self.execution_mode == model_fn_lib.ModeKeys.TRAIN)
948 is_training = self.execution_mode == model_fn_lib.ModeKeys.TRAIN
949 is_test = self.execution_mode == model_fn_lib.ModeKeys.EVAL
950 is_predict = self.execution_mode == model_fn_lib.ModeKeys.PREDICT
1211 if (self.execution_mode == model_fn_lib.ModeKeys.TRAIN or
1212 self.execution_mode == model_fn_lib.ModeKeys.EVAL):
[all …]
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dgmm.py28 from tensorflow.contrib.learn.python.learn.estimators import model_fn as model_fn_lib unknown
178 return model_fn_lib.ModelFnOps(mode=mode, predictions=predictions,
Dkmeans.py27 from tensorflow.python.estimator import model_fn as model_fn_lib unknown
231 return model_fn_lib.EstimatorSpec(
/external/tensorflow/tensorflow/contrib/distribute/python/
Dvalues_test.py30 from tensorflow.python.estimator import model_fn as model_fn_lib unknown
320 spec = model_fn_lib.EstimatorSpec(
321 mode=model_fn_lib.ModeKeys.TRAIN,
332 isinstance(merged_estimator_spec, model_fn_lib.EstimatorSpec))
333 self.assertEqual(model_fn_lib.ModeKeys.TRAIN, merged_estimator_spec.mode)