/external/tensorflow/tensorflow/python/estimator/export/ |
H A D | export_lib.py | 25 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 D | export_output_test.py | 24 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 D | export_test.py | 29 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 D | export.py | 203 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 D | core_estimator_predictor.py | 40 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 D | testing_common.py | 29 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 D | kmeans.py | 28 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 D | replicate_model_fn.py | 36 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 D | head.py | 25 from tensorflow.python.estimator.export import export_output namespace 469 export_output.PredictOutput(predictions))
|
H A D | replicate_model_fn_test.py | 34 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 D | model_fn_test.py | 23 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 D | model_fn.py | 26 from tensorflow.python.estimator.export.export_output import ExportOutput
|
H A D | estimator_test.py | 38 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 D | head.py | 30 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 D | model_fn.py | 31 from tensorflow.python.estimator.export import export_output as core_export_lib
|
H A D | model_fn_test.py | 26 from tensorflow.python.estimator.export import export_output as core_export_lib
|
/external/tensorflow/tensorflow/python/keras/_impl/keras/ |
H A D | estimator.py | 206 export_lib.export_output.PredictOutput(predictions)
|