Home
last modified time | relevance | path

Searched refs:INFER (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dmodel_fn_test.py48 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 …]
Dmodel_fn.py62 INFER = 'infer' variable in ModeKeys
66 if key not in (cls.TRAIN, cls.EVAL, cls.INFER):
170 if mode == ModeKeys.INFER or mode == ModeKeys.EVAL:
295 elif self.mode == ModeKeys.INFER:
Dstate_saving_rnn_estimator.py186 if mode != model_fn.ModeKeys.INFER:
489 if mode != model_fn.ModeKeys.INFER:
504 if mode != model_fn.ModeKeys.INFER:
509 if mode != model_fn.ModeKeys.INFER:
Ddynamic_rnn_estimator.py505 if mode != model_fn.ModeKeys.INFER:
512 if mode != model_fn.ModeKeys.INFER:
519 if mode != model_fn.ModeKeys.INFER:
Dhead.py655 if (mode != model_fn.ModeKeys.INFER) and (labels is not None):
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:
1670 if mode == model_fn.ModeKeys.INFER:
1746 mode=model_fn.ModeKeys.INFER,
1855 if (mode != model_fn.ModeKeys.TRAIN and mode != model_fn.ModeKeys.INFER and
Destimator_input_test.py112 model_fn.ModeKeys.INFER)
126 model_fn.ModeKeys.INFER)
141 model_fn.ModeKeys.INFER)
Dstate_saving_rnn_estimator_test.py372 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),
Ddynamic_rnn_estimator_test.py152 mode != model_fn_lib.ModeKeys.INFER else None)
241 model_fn_ops = self._GetModelFnOpsForMode(model_fn_lib.ModeKeys.INFER)
273 if mode == model_fn_lib.ModeKeys.INFER:
305 input_fn=get_input_fn(model_fn_lib.ModeKeys.INFER),
Dhead_test.py541 {}, model_fn.ModeKeys.INFER, self._labels, head_lib.no_op_train_fn,
846 {}, model_fn.ModeKeys.INFER, self._labels, head_lib.no_op_train_fn,
876 {}, model_fn.ModeKeys.INFER, self._labels, head_lib.no_op_train_fn,
1344 mode=model_fn.ModeKeys.INFER,
1396 mode=model_fn.ModeKeys.INFER,
1797 mode=model_fn.ModeKeys.INFER,
Destimator_test.py131 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:
Destimator.py1314 return self._call_model_fn(features, labels, model_fn_lib.ModeKeys.INFER)
1399 model_fn_lib.ModeKeys.INFER)
/external/tensorflow/tensorflow/contrib/learn/python/learn/utils/
Dsaved_model_export_utils_test.py378 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,
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dwals_test.py161 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:
291 mode=model_fn.ModeKeys.INFER,
308 mode=model_fn.ModeKeys.INFER,
Dwals.py318 elif mode == model_fn.ModeKeys.INFER:
342 mode=model_fn.ModeKeys.INFER,
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Destimator_utils.py31 model_fn_lib.ModeKeys.PREDICT: contrib_model_fn_lib.ModeKeys.INFER
Ddnn_tree_combined_estimator.py221 if predict_with_tree_only and mode == model_fn.ModeKeys.INFER:
305 if mode == model_fn.ModeKeys.TRAIN or mode == model_fn.ModeKeys.INFER:
Dmodel.py317 if mode == learn.ModeKeys.INFER:
/external/tensorflow/tensorflow/contrib/tensor_forest/client/
Drandom_forest.py198 if (mode == model_fn_lib.ModeKeys.INFER or
512 mode == model_fn_lib.ModeKeys.INFER):
/external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/
Dgbdt_batch.py453 output_leaf_index_modes = [learn.ModeKeys.INFER]
456 learn.ModeKeys.INFER) for mode in output_leaf_index_modes):
Dgbdt_batch_test.py1038 mode = model_fn.ModeKeys.INFER