| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| model_fn_test.py | 48 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...] |
| model_fn.py | 57 * `INFER`: inference mode.
62 INFER = 'infer'
66 if key not in (cls.TRAIN, cls.EVAL, cls.INFER):
170 if mode == ModeKeys.INFER or mode == ModeKeys.EVAL:
219 used if a Servo request does not explicitly mention which head to infer
295 elif self.mode == ModeKeys.INFER:
|
| state_saving_rnn_estimator.py | 186 if mode != model_fn.ModeKeys.INFER:
294 input_length=None, # infer sequence lengths
489 if mode != model_fn.ModeKeys.INFER:
504 if mode != model_fn.ModeKeys.INFER:
509 if mode != model_fn.ModeKeys.INFER:
|
| dynamic_rnn_estimator.py | 505 if mode != model_fn.ModeKeys.INFER:
512 if mode != model_fn.ModeKeys.INFER:
519 if mode != model_fn.ModeKeys.INFER:
|
| estimator_input_test.py | 112 model_fn.ModeKeys.INFER)
126 model_fn.ModeKeys.INFER)
141 model_fn.ModeKeys.INFER)
|
| state_saving_rnn_estimator_test.py | 326 """Helper for testGetRnnModelFn{Train,Eval,Infer}()."""
353 # testGetRnnModelFn{Train,Eval,Infer}() test which fields
372 model_fn_ops = self._getModelFnOpsForMode(model_fn_lib.ModeKeys.INFER)
404 if mode == model_fn_lib.ModeKeys.INFER:
439 model_fn_lib.ModeKeys.INFER, seed=4321),
|
| dynamic_rnn_estimator_test.py | 152 mode != model_fn_lib.ModeKeys.INFER else None)
222 # testGetDynamicRnnModelFn{Train,Eval,Infer}() test which fields
241 model_fn_ops = self._GetModelFnOpsForMode(model_fn_lib.ModeKeys.INFER)
249 """Helper for testGetDynamicRnnModelFn{Train,Eval,Infer}()."""
273 if mode == model_fn_lib.ModeKeys.INFER:
305 input_fn=get_input_fn(model_fn_lib.ModeKeys.INFER),
|
| head.py | 655 if (mode != model_fn.ModeKeys.INFER) and (labels is not None):
762 if (mode == model_fn.ModeKeys.INFER) or (labels is None):
[all...] |
| head_test.py | 541 {}, model_fn.ModeKeys.INFER, self._labels, head_lib.no_op_train_fn,
[all...] |
| estimator_test.py | 131 model_fn.ModeKeys.INFER)
145 model_fn.ModeKeys.INFER)
160 model_fn.ModeKeys.INFER)
246 model_fn.ModeKeys.INFER)
262 if mode == model_fn.ModeKeys.INFER:
[all...] |
| estimator.py | [all...] |
| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| estimator_utils.py | 31 model_fn_lib.ModeKeys.PREDICT: contrib_model_fn_lib.ModeKeys.INFER
|
| dnn_tree_combined_estimator.py | 95 mode: Mode we are in. (TRAIN/EVAL/INFER)
221 if predict_with_tree_only and mode == model_fn.ModeKeys.INFER:
305 if mode == model_fn.ModeKeys.TRAIN or mode == model_fn.ModeKeys.INFER:
[all...] |
| model.py | 49 mode: Mode we are in. (TRAIN/EVAL/INFER)
207 mode: Mode we are in. (TRAIN/EVAL/INFER)
317 if mode == learn.ModeKeys.INFER:
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/ |
| saved_model_export_utils_test.py | 378 model_fn.ModeKeys.INFER,
395 model_fn.ModeKeys.INFER,
414 model_fn.ModeKeys.INFER,
435 model_fn.ModeKeys.INFER,
449 model_fn.ModeKeys.INFER,
465 model_fn.ModeKeys.INFER,
478 model_fn.ModeKeys.INFER,
494 model_fn.ModeKeys.INFER,
531 model_fn.ModeKeys.INFER,
596 model_fn.ModeKeys.INFER,
[all...] |
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| wals_test.py | 96 In INFER and EVAL modes, one must also provide project_row, a boolean which
102 mode: Can be one of model_fn.ModeKeys.{TRAIN, INFER, EVAL}.
103 project_row: A boolean. Used in INFER and EVAL modes. Specifies whether
105 projection_weights: A float numpy array. Used in INFER mode. Specifies
161 if mode == model_fn.ModeKeys.INFER or mode == model_fn.ModeKeys.EVAL:
164 msg='project_row must be specified in INFER or EVAL mode.')
168 if mode == model_fn.ModeKeys.INFER and projection_weights is not None:
291 mode=model_fn.ModeKeys.INFER,
308 mode=model_fn.ModeKeys.INFER,
|
| wals.py | 317 # INFER mode
318 elif mode == model_fn.ModeKeys.INFER:
342 mode=model_fn.ModeKeys.INFER,
426 INFER:
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/client/ |
| random_forest.py | 196 # Also ignore device assigner if we're exporting (mode == INFER)
198 if (mode == model_fn_lib.ModeKeys.INFER or
512 mode == model_fn_lib.ModeKeys.INFER):
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| gbdt_batch.py | 275 mode: Train/Eval/Infer
328 output_leaf_index_modes: A list of modes from (TRAIN, EVAL, INFER) which
330 are only outputted in INFER mode.
453 output_leaf_index_modes = [learn.ModeKeys.INFER]
456 learn.ModeKeys.INFER) for mode in output_leaf_index_modes):
468 mode: learn.ModeKeys.TRAIN or EVAL or INFER.
[all...] |
| gbdt_batch_test.py | [all...] |
