| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| D | head_test.py | 61 model_fn = _stub_model_fn()
65 model_fn(features={}, labels={"a": "b"}, mode=mode)
68 model_fn(features={}, labels=array_ops.zeros([]), mode=mode)
71 model_fn = _stub_model_fn()
73 model_fn(features={}, labels={}, mode="Not a mode")
115 model_fn = ts_head_lib.TimeSeriesRegressionHead(
119 outputs = model_fn(
200 model_fn = _stub_model_fn()
204 model_fn(
210 model_fn = _stub_model_fn()
[all …]
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | mirrored_strategy_multigpu_test.py | 229 def model_fn(): function
251 result = distribution.extended.call_for_each_replica(model_fn)
265 def model_fn(): function
271 in_model_fn = distribution.extended.call_for_each_replica(model_fn)
279 in_model_fn = distribution.extended.call_for_each_replica(model_fn)
286 def model_fn(): function
290 result = distribution.extended.call_for_each_replica(model_fn)
298 def model_fn(): function
305 distribution.extended.call_for_each_replica(model_fn)
325 def model_fn(): function
[all …]
|
| D | minimize_loss_test.py | 61 model_fn, dataset_fn, layer = minimize_loss_example(
68 model_fn, args=(inputs,)))
99 model_fn, dataset_fn, layer = minimize_loss_example(
107 model_fn, args=(iterator.get_next(),)))
149 model_fn, dataset_fn, _ = minimize_loss_example(
159 model_fn, args=(inputs,)))
214 model_fn, dataset_fn, batchnorm = batchnorm_example(
225 model_fn, args=(inputs,)))
294 def model_fn(inputs): function
320 model_fn, args=(inputs,)))
[all …]
|
| D | single_loss_example.py | 72 def model_fn(x): function
86 return model_fn, dataset_fn, layer
109 def model_fn(x): function
125 return model_fn, dataset_fn, batchnorm
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | estimator_input_test.py | 33 from tensorflow.contrib.learn.python.learn.estimators import model_fn
111 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)
148 return model_fn.ModelFnOps(
177 est = estimator.Estimator(model_fn=linear_model_fn, model_dir=output_dir)
189 est2 = estimator.Estimator(model_fn=linear_model_fn, model_dir=output_dir)
[all …]
|
| D | head_test.py | 29 from tensorflow.contrib.learn.python.learn.estimators import model_fn
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,
224 mode=model_fn.ModeKeys.EVAL,
240 mode=model_fn.ModeKeys.TRAIN,
256 mode=model_fn.ModeKeys.TRAIN,
[all …]
|
| D | estimator_test.py | 46 from tensorflow.contrib.learn.python.learn.estimators import model_fn
130 assert mode in (model_fn.ModeKeys.TRAIN, model_fn.ModeKeys.EVAL,
131 model_fn.ModeKeys.INFER)
144 assert mode in (model_fn.ModeKeys.TRAIN, model_fn.ModeKeys.EVAL,
145 model_fn.ModeKeys.INFER)
159 assert mode in (model_fn.ModeKeys.TRAIN, model_fn.ModeKeys.EVAL,
160 model_fn.ModeKeys.INFER)
167 return model_fn.ModelFnOps(
245 assert mode in (model_fn.ModeKeys.TRAIN, model_fn.ModeKeys.EVAL,
246 model_fn.ModeKeys.INFER)
[all …]
|
| D | estimators_test.py | 59 def model_fn(features, labels): function
69 model_fn=model_fn, feature_engineering_fn=feature_engineering_fn)
99 def model_fn(features, labels): function
109 model_fn=model_fn, feature_engineering_fn=feature_engineering_fn)
140 def model_fn(features, labels): function
150 model_fn=model_fn, feature_engineering_fn=feature_engineering_fn)
152 estimator_without_fe_fn = estimator_lib.Estimator(model_fn=model_fn)
|
| D | model_fn_test.py | 24 from tensorflow.contrib.learn.python.learn.estimators import model_fn
48 mode=model_fn.ModeKeys.INFER):
50 return model_fn.ModelFnOps(
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)
[all …]
|
| D | logistic_regressor.py | 33 from tensorflow.contrib.learn.python.learn.estimators import model_fn as model_fn_lib
38 def _get_model_fn_with_logistic_metrics(model_fn): argument
54 predictions, loss, train_op = model_fn(features, labels, mode)
79 model_fn, thresholds=None, model_dir=None, config=None, argument
126 model_fn=_get_model_fn_with_logistic_metrics(model_fn),
|
| D | state_saving_rnn_estimator.py | 32 from tensorflow.contrib.learn.python.learn.estimators import model_fn
186 if mode != model_fn.ModeKeys.INFER:
471 if mode == model_fn.ModeKeys.TRAIN
489 if mode != model_fn.ModeKeys.INFER:
504 if mode != model_fn.ModeKeys.INFER:
509 if mode != model_fn.ModeKeys.INFER:
518 if mode == model_fn.ModeKeys.TRAIN:
527 return model_fn.ModelFnOps(mode=mode,
661 model_fn=rnn_model_fn,
|
| D | dynamic_rnn_estimator.py | 30 from tensorflow.contrib.learn.python.learn.estimators import model_fn
487 if mode == model_fn.ModeKeys.TRAIN
505 if mode != model_fn.ModeKeys.INFER:
512 if mode != model_fn.ModeKeys.INFER:
519 if mode != model_fn.ModeKeys.INFER:
525 if mode == model_fn.ModeKeys.TRAIN:
538 return model_fn.ModelFnOps(mode=mode,
699 model_fn=dynamic_rnn_model_fn,
|
| D | head.py | 33 from tensorflow.contrib.learn.python.learn.estimators import model_fn
655 if (mode != model_fn.ModeKeys.INFER) and (labels is not None):
664 if mode == model_fn.ModeKeys.TRAIN:
672 return model_fn.ModelFnOps(
762 if (mode == model_fn.ModeKeys.INFER) or (labels is None):
884 if (mode == model_fn.ModeKeys.INFER) or (labels is None):
1089 if (mode == model_fn.ModeKeys.INFER) or (labels is None):
1286 if (mode == model_fn.ModeKeys.INFER) or (labels is None):
1386 if (mode == model_fn.ModeKeys.INFER) or (labels is None):
1504 if mode != model_fn.ModeKeys.INFER:
[all …]
|
| /external/tensorflow/tensorflow/python/estimator/ |
| D | BUILD | 28 ":model_fn",
60 name = "model_fn",
61 srcs = ["model_fn.py"],
104 ":model_fn",
120 ":model_fn",
133 ":model_fn",
146 ":model_fn",
163 ":model_fn",
183 ":model_fn",
211 ":model_fn",
[all …]
|
| D | model_fn.py | 26 from tensorflow_estimator.python.estimator import model_fn
30 model_fn.__all__ = [s for s in dir(model_fn) if not s.startswith('__')]
32 from tensorflow_estimator.python.estimator.model_fn import *
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | wals.py | 23 from tensorflow.contrib.learn.python.learn.estimators import model_fn
160 and mode == model_fn.ModeKeys.TRAIN)
162 mode == model_fn.ModeKeys.TRAIN)
185 if mode == model_fn.ModeKeys.TRAIN:
309 return model_fn.ModelFnOps(
310 mode=model_fn.ModeKeys.TRAIN,
318 elif mode == model_fn.ModeKeys.INFER:
341 return model_fn.ModelFnOps(
342 mode=model_fn.ModeKeys.INFER,
350 elif mode == model_fn.ModeKeys.EVAL:
[all …]
|
| D | wals_test.py | 28 from tensorflow.contrib.learn.python.learn.estimators import model_fn
161 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,
[all …]
|
| /external/tensorflow/tensorflow/contrib/gan/python/estimator/python/ |
| D | latent_gan_estimator_impl.py | 76 from tensorflow.python.estimator import model_fn as model_fn_lib
96 def model_fn(features, labels, mode, params): function
135 return model_fn
187 model_fn = _get_latent_gan_model_fn(generator_fn, discriminator_fn,
204 return estimator.Estimator(model_fn=model_fn, config=config, params=params,
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/ |
| D | saved_model_export_utils_test.py | 28 from tensorflow.contrib.learn.python.learn.estimators import model_fn
377 model_fn_ops = model_fn.ModelFnOps(
378 model_fn.ModeKeys.INFER,
394 model_fn_ops = model_fn.ModelFnOps(
395 model_fn.ModeKeys.INFER,
413 model_fn_ops = model_fn.ModelFnOps(
414 model_fn.ModeKeys.INFER,
434 model_fn_ops = model_fn.ModelFnOps(
435 model_fn.ModeKeys.INFER,
448 model_fn_ops = model_fn.ModelFnOps(
[all …]
|
| /external/tensorflow/tensorflow/contrib/estimator/ |
| D | BUILD | 57 "//tensorflow/python/estimator:model_fn",
72 "//tensorflow/python/estimator:model_fn",
86 "//tensorflow/python/estimator:model_fn",
115 "//tensorflow/python/estimator:model_fn",
159 "//tensorflow/python/estimator:model_fn",
174 "//tensorflow/python/estimator:model_fn",
216 "//tensorflow/python/estimator:model_fn",
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | checkpoint_input_pipeline_hook_test.py | 34 from tensorflow_estimator.python.estimator import model_fn
52 return model_fn.EstimatorSpec(
77 est = estimator.Estimator(model_fn=self._model_fn)
91 est = estimator.Estimator(model_fn=self._model_fn)
103 est = estimator.Estimator(model_fn=self._model_fn)
118 est = estimator.Estimator(model_fn=self._model_fn)
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| D | estimator_utils.py | 22 from tensorflow.contrib.learn.python.learn.estimators import model_fn
23 from tensorflow.contrib.learn.python.learn.estimators import model_fn as contrib_model_fn_lib
25 from tensorflow.python.estimator import model_fn as model_fn_lib
68 return model_fn.ModelFnOps(
|
| D | dnn_tree_combined_estimator.py | 38 from tensorflow.contrib.learn.python.learn.estimators import model_fn
207 if dnn_dropout is not None and mode == model_fn.ModeKeys.TRAIN:
221 if predict_with_tree_only and mode == model_fn.ModeKeys.INFER:
224 elif predict_with_tree_only and mode == model_fn.ModeKeys.EVAL:
305 if mode == model_fn.ModeKeys.TRAIN or mode == model_fn.ModeKeys.INFER:
331 if mode != model_fn.ModeKeys.TRAIN:
363 if mode != model_fn.ModeKeys.TRAIN:
511 model_fn=_model_fn,
642 model_fn=_model_fn,
758 model_fn=_model_fn,
[all …]
|
| D | estimator.py | 128 model_fn=model.model_builder,
221 model_fn=model.model_builder,
303 model_fn=model.model_builder,
394 model_fn=model.ranking_model_builder,
498 model_fn=model.model_builder,
652 model_fn=_model_fn, model_dir=model_dir, config=config)
735 model_fn=_model_fn, model_dir=model_dir, config=config)
837 model_fn=_model_fn, model_dir=model_dir, config=config)
|
| /external/tensorflow/tensorflow/contrib/predictor/ |
| D | core_estimator_predictor.py | 23 from tensorflow.python.estimator import model_fn
35 estimator_spec = estimator.model_fn(
36 serving_input_receiver.features, None, model_fn.ModeKeys.PREDICT,
|