/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
estimator_utils.py | 25 from tensorflow.python.estimator import model_fn as model_fn_lib 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/contrib/gan/python/estimator/python/ |
stargan_estimator_impl.py | 29 from tensorflow.python.estimator import model_fn as model_fn_lib 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: 199 else: # model_fn_lib.ModeKeys.TRAIN or model_fn_lib.ModeKeys.EVAL 215 if mode == model_fn_lib.ModeKeys.PREDICT: 216 estimator_spec = model_fn_lib.EstimatorSpec [all...] |
head_impl.py | 25 from tensorflow.python.estimator import model_fn as model_fn_lib 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 [all...] |
gan_estimator_impl.py | 29 from tensorflow.python.estimator import model_fn as model_fn_lib 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: 210 else: # model_fn_lib.ModeKeys.TRAIN or model_fn_lib.ModeKeys.EVAL 223 if mode == model_fn_lib.ModeKeys.PREDICT: 224 estimator_spec = model_fn_lib.EstimatorSpec( 232 if mode == model_fn_lib.ModeKeys.EVAL [all...] |
head_test.py | 24 from tensorflow.python.estimator import model_fn as model_fn_lib 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)
|
tpu_gan_estimator_impl.py | 27 from tensorflow.python.estimator import model_fn as model_fn_lib 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: 295 else: # model_fn_lib.ModeKeys.TRAIN: 358 mode=model_fn_lib.ModeKeys.EVAL [all...] |
stargan_estimator_test.py | 30 from tensorflow.python.estimator import model_fn as model_fn_lib 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...] |
gan_estimator_test.py | 34 from tensorflow.python.estimator import model_fn as model_fn_lib 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...] |
tpu_gan_estimator_test.py | 35 from tensorflow.python.estimator import model_fn as model_fn_lib 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:
|
latent_gan_estimator_impl.py | 76 from tensorflow.python.estimator import model_fn as model_fn_lib 131 return model_fn_lib.EstimatorSpec(mode=mode,
|
/external/tensorflow/tensorflow/python/tpu/ |
_tpu_estimator_embedding.py | 23 from tensorflow.python.estimator import model_fn as model_fn_lib 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)
|
tpu_context.py | 24 from tensorflow.python.estimator import model_fn as model_fn_lib 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...] |
tpu_estimator.py | 42 from tensorflow.python.estimator import model_fn as model_fn_lib 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: [all...] |
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
logistic_regressor.py | 33 from tensorflow.contrib.learn.python.learn.estimators import model_fn as model_fn_lib 55 if mode == model_fn_lib.ModeKeys.EVAL: 62 return model_fn_lib.ModelFnOps(
|
state_saving_rnn_estimator_test.py | 29 from tensorflow.contrib.learn.python.learn.estimators import model_fn as model_fn_lib 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),
|
composable_model_test.py | 27 from tensorflow.contrib.learn.python.learn.estimators import model_fn as model_fn_lib 50 if mode == model_fn_lib.ModeKeys.TRAIN: 52 elif mode == model_fn_lib.ModeKeys.EVAL:
|
dynamic_rnn_estimator_test.py | 30 from tensorflow.contrib.learn.python.learn.estimators import model_fn as model_fn_lib 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),
|
estimator.py | 48 from tensorflow.contrib.learn.python.learn.estimators import model_fn as model_fn_lib [all...] |
/external/tensorflow/tensorflow/contrib/tensor_forest/client/ |
random_forest.py | 25 from tensorflow.contrib.learn.python.learn.estimators import model_fn as model_fn_lib 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 [all...] |
/external/tensorflow/tensorflow/contrib/compiler/ |
xla.py | 29 from tensorflow.python.estimator import model_fn as model_fn_lib 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:
|
xla_test.py | 30 from tensorflow.python.estimator import model_fn as model_fn_lib 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/factorization/python/ops/ |
gmm.py | 28 from tensorflow.contrib.learn.python.learn.estimators import model_fn as model_fn_lib 178 return model_fn_lib.ModelFnOps(mode=mode, predictions=predictions,
|
kmeans.py | 27 from tensorflow.python.estimator import model_fn as model_fn_lib 231 return model_fn_lib.EstimatorSpec(
|
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
keras_support.py | 71 from tensorflow.python.estimator import model_fn as model_fn_lib [all...] |
/external/tensorflow/tensorflow/contrib/distribute/python/ |
values_test.py | 30 from tensorflow.python.estimator import model_fn as model_fn_lib 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)
|