Searched refs:export_output (Results 1 - 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/python/estimator/export/
H A Dexport_lib.py25 from tensorflow.python.estimator.export.export_output import ClassificationOutput
26 from tensorflow.python.estimator.export.export_output import ExportOutput
27 from tensorflow.python.estimator.export.export_output import PredictOutput
28 from tensorflow.python.estimator.export.export_output import RegressionOutput
H A Dexport_output_test.py24 from tensorflow.python.estimator.export import export_output as export_output_lib
73 export_output = export_output_lib.RegressionOutput(value)
74 actual_signature_def = export_output.as_signature_def(input_tensors)
104 export_output = export_output_lib.ClassificationOutput(classes=classes)
105 actual_signature_def = export_output.as_signature_def(input_tensors)
138 export_output = export_output_lib.ClassificationOutput(
140 actual_signature_def = export_output.as_signature_def(input_tensors)
178 export_output = export_output_lib.ClassificationOutput(
180 actual_signature_def = export_output.as_signature_def(input_tensors)
H A Dexport_test.py29 from tensorflow.python.estimator.export import export_output namespace
227 export_output.RegressionOutput(value=output_1),
228 "head-2": export_output.ClassificationOutput(classes=output_2),
229 "head-3": export_output.PredictOutput(outputs={
263 export_output.RegressionOutput(value=output_1),
264 "head-2": export_output.ClassificationOutput(classes=output_2),
265 "head-3": export_output.PredictOutput(outputs={
315 export_output.RegressionOutput(value=output_1),
316 "head-2": export_output.ClassificationOutput(classes=output_2),
317 "head-3": export_output
[all...]
H A Dexport.py203 for output_key, export_output in export_outputs.items():
206 signature = export_output.as_signature_def(receiver_tensors)
217 for output_key, export_output in export_outputs.items():
221 signature = export_output.as_signature_def(receiver_tensors_alt)
228 # The above calls to export_output.as_signature_def should return only
/external/tensorflow/tensorflow/contrib/predictor/
H A Dcore_estimator_predictor.py40 export_output = export_outputs.get(output_key)
41 if export_output is None:
44 return export_output.as_signature_def(serving_input_receiver.receiver_tensors)
H A Dtesting_common.py29 from tensorflow.python.estimator.export import export_output namespace
57 export_outputs = {k: export_output.PredictOutput({k: v})
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
H A Dkmeans.py28 from tensorflow.python.estimator.export import export_output namespace
214 export_output.PredictOutput(all_distances[0]),
216 export_output.PredictOutput(model_predictions[0]),
218 export_output.PredictOutput(model_predictions[0])
/external/tensorflow/tensorflow/contrib/estimator/python/estimator/
H A Dreplicate_model_fn.py36 from tensorflow.python.estimator.export import export_output as export_output_lib
752 export_output.outputs for export_output in export_output_list
758 [export_output.value for export_output in export_output_list],
765 [export_output.scores for export_output in export_output_list],
771 [export_output.classes for export_output in export_output_list],
H A Dhead.py25 from tensorflow.python.estimator.export import export_output namespace
469 export_output.PredictOutput(predictions))
H A Dreplicate_model_fn_test.py34 from tensorflow.python.estimator.export import export_output namespace
1448 export_output.PredictOutput(predictions),
1450 export_output.ClassificationOutput(predictions['probabilities'],
1453 export_output.ClassificationOutput(
1456 export_output.ClassificationOutput(classes=classes),
1458 export_output.RegressionOutput(predictions['probabilities']),
/external/tensorflow/tensorflow/python/estimator/
H A Dmodel_fn_test.py23 from tensorflow.python.estimator.export import export_output namespace
70 'head_name': export_output.ClassificationOutput(classes=classes)
222 'head_name': export_output.ClassificationOutput(classes=classes)
464 'head_name': export_output.ClassificationOutput(classes=classes)
520 export_outputs=export_output.ClassificationOutput(classes=classes))
539 regression_output = export_output.RegressionOutput(value=output_1)
562 export_output.RegressionOutput(value=output_1),
563 'head-2': export_output.ClassificationOutput(classes=output_2),
564 'head-3': export_output.PredictOutput(outputs={
580 'head-1': export_output
[all...]
H A Dmodel_fn.py26 from tensorflow.python.estimator.export.export_output import ExportOutput
H A Destimator_test.py38 from tensorflow.python.estimator.export import export_output namespace
1726 'test': export_output.ClassificationOutput(scores, classes)})
1742 'test': export_output.PredictOutput({'prediction': prediction})})
1955 export_outputs={'test': export_output.ClassificationOutput(scores)})
1993 export_outputs={'test': export_output.ClassificationOutput(scores)})
2031 'test': export_output.ClassificationOutput(
2054 'test': export_output.ClassificationOutput(
2090 'test': export_output.PredictOutput({
2241 'predictions': export_output.RegressionOutput(predictions)
/external/tensorflow/tensorflow/python/estimator/canned/
H A Dhead.py30 from tensorflow.python.estimator.export import export_output namespace
450 return export_output.ClassificationOutput(
757 _PREDICT_SERVING_KEY: export_output.PredictOutput(predictions)
1093 _REGRESS_SERVING_KEY: export_output.RegressionOutput(
1095 _PREDICT_SERVING_KEY: export_output.PredictOutput(predictions)
1299 regression_output = export_output.RegressionOutput(value=logits)
1306 _PREDICT_SERVING_KEY: export_output.PredictOutput(predictions)
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
H A Dmodel_fn.py31 from tensorflow.python.estimator.export import export_output as core_export_lib
H A Dmodel_fn_test.py26 from tensorflow.python.estimator.export import export_output as core_export_lib
/external/tensorflow/tensorflow/python/keras/_impl/keras/
H A Destimator.py206 export_lib.export_output.PredictOutput(predictions)

Completed in 1220 milliseconds