Searched refs:output_alternatives (Results 1 – 14 of 14) sorted by relevance
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
D | model_fn_test.py | 47 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 …]
|
D | model_fn.py | 92 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()}
|
D | head_test.py | 104 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 …]
|
D | dynamic_rnn_estimator.py | 534 output_alternatives = _get_output_alternatives(prediction_type, 543 output_alternatives=output_alternatives)
|
D | logistic_regressor.py | 68 output_alternatives={
|
D | head.py | 678 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)
|
D | estimator.py | 1400 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/ |
D | saved_model_export_utils.py | 215 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 …]
|
D | saved_model_export_utils_test.py | 380 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/ |
D | testing_common.py | 67 output_alternatives = {k: (constants.ProblemType.UNSPECIFIED, {k: v}) 72 output_alternatives=output_alternatives,
|
D | contrib_estimator_predictor.py | 74 (output_alternatives, 77 _, fetch_tensors = output_alternatives[output_alternative_key]
|
/external/tensorflow/tensorflow/contrib/tensor_forest/client/ |
D | random_forest.py | 286 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/ |
D | estimator_utils.py | 74 output_alternatives=alternatives)
|
/external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/ |
D | estimator_test.py | 44 k: v[0] for k, v in six.iteritems(model_fn_ops.output_alternatives)
|