Home
last modified time | relevance | path

Searched refs:output_alternatives (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dmodel_fn_test.py47 def create_model_fn_ops(self, predictions, output_alternatives, argument
61 output_alternatives=output_alternatives,
91 output_alternatives = {"regression_head": (
94 predictions, output_alternatives, mode=model_fn.ModeKeys.INFER)
110 output_alternatives = {"regression_head": (
114 predictions, output_alternatives, mode=model_fn.ModeKeys.INFER)
128 output_alternatives = {"classification_head": (
131 predictions, output_alternatives, mode=model_fn.ModeKeys.INFER)
150 output_alternatives = {"classification_head": (
153 predictions, output_alternatives, mode=model_fn.ModeKeys.INFER)
[all …]
Dmodel_fn.py92 output_alternatives=None, argument
206 output_alternatives=output_alternatives,
268 if self.output_alternatives:
269 output_alternatives = self.output_alternatives
271 if (len(output_alternatives) > 1 and
273 output_alternatives):
274 output_alternatives = output_alternatives.copy()
275 output_alternatives[
277 output_alternatives[default_serving_output_alternative_key])
279 output_alternatives.items()}
Dhead_test.py104 k: v[0] for k, v in six.iteritems(model_fn_ops.output_alternatives)
139 k: v[0] for k, v in six.iteritems(model_fn_ops.output_alternatives)
347 k: v[0] for k, v in six.iteritems(model_fn_ops.output_alternatives)
549 list(model_fn_ops.output_alternatives))
552 model_fn_ops.output_alternatives["head_name"][0])
555 model_fn_ops.output_alternatives["head_name"][1])
737 k: v[0] for k, v in six.iteritems(model_fn_ops.output_alternatives)
854 list(model_fn_ops.output_alternatives))
857 model_fn_ops.output_alternatives["head_name"][0])
859 model_fn_ops.output_alternatives["head_name"][1])
[all …]
Ddynamic_rnn_estimator.py534 output_alternatives = _get_output_alternatives(prediction_type,
543 output_alternatives=output_alternatives)
Dlogistic_regressor.py68 output_alternatives={
Dhead.py678 output_alternatives=create_output_alternatives_fn(predictions))
1736 output_alternatives = {}
1741 output_alternatives[head_name] = m.output_alternatives[head_name]
1748 output_alternatives=output_alternatives)
Destimator.py1400 output_alternatives, actual_default_output_alternative_key = (
1411 input_alternatives, output_alternatives,
1457 for output_alternative in output_alternatives.values()
/external/tensorflow/tensorflow/contrib/learn/python/learn/utils/
Dsaved_model_export_utils.py215 output_alternatives = model_fn_ops.output_alternatives
217 if not output_alternatives:
231 output_alternatives = {
235 return output_alternatives, actual_default_output_alternative_key
239 if default_output_alternative_key in output_alternatives:
240 return output_alternatives, default_output_alternative_key
245 sorted(output_alternatives.keys())))
247 if len(output_alternatives) == 1:
249 (actual_default_output_alternative_key, _), = output_alternatives.items()
250 return output_alternatives, actual_default_output_alternative_key
[all …]
Dsaved_model_export_utils_test.py380 output_alternatives=provided_output_alternatives)
382 output_alternatives, _ = saved_model_export_utils.get_output_alternatives(
385 self.assertEqual(provided_output_alternatives, output_alternatives)
397 output_alternatives=provided_output_alternatives)
416 output_alternatives=provided_output_alternatives)
418 output_alternatives, _ = saved_model_export_utils.get_output_alternatives(
425 }, output_alternatives)
437 output_alternatives=provided_output_alternatives)
451 output_alternatives=None)
453 output_alternatives, _ = saved_model_export_utils.get_output_alternatives(
[all …]
/external/tensorflow/tensorflow/contrib/predictor/
Dtesting_common.py67 output_alternatives = {k: (constants.ProblemType.UNSPECIFIED, {k: v})
72 output_alternatives=output_alternatives,
Dcontrib_estimator_predictor.py74 (output_alternatives,
77 _, fetch_tensors = output_alternatives[output_alternative_key]
/external/tensorflow/tensorflow/contrib/tensor_forest/client/
Drandom_forest.py286 if not model_ops.output_alternatives:
287 model_ops.output_alternatives = {}
288 model_ops.output_alternatives[ALL_SERVING_KEY] = (
538 output_alternatives=None)
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Destimator_utils.py74 output_alternatives=alternatives)
/external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/
Destimator_test.py44 k: v[0] for k, v in six.iteritems(model_fn_ops.output_alternatives)