Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:model_fn_lib (Results 1 - 25 of 25) sorted by null

/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)

Completed in 1964 milliseconds