Home
last modified time | relevance | path

Searched refs:classification_signature (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/contrib/session_bundle/
Dbundle_shim_test.py180 classification_signature = manifest_pb2.ClassificationSignature()
181 classification_signature.input.CopyFrom(
184 classification_signature.classes.CopyFrom(
187 classification_signature.scores.CopyFrom(
190 signatures_proto.default_signature.classification_signature.CopyFrom(
191 classification_signature)
193 signatures_proto.default_signature.classification_signature.CopyFrom(
194 classification_signature)
225 classification_signature = manifest_pb2.ClassificationSignature()
227 signature_constants.PREDICT_INPUTS].classification_signature.CopyFrom(
[all …]
Dbundle_shim.py97 classification_signature = default_signature.classification_signature
99 _add_input_to_signature_def(classification_signature.input.tensor_name,
102 _add_output_to_signature_def(classification_signature.classes.tensor_name,
105 _add_output_to_signature_def(classification_signature.scores.tensor_name,
Dbundle_shim.cc64 const ClassificationSignature& classification_signature, in BuildClassificationSignatureDef() argument
69 classification_signature.input().tensor_name(), tensor_name_to_dtype, in BuildClassificationSignatureDef()
72 classification_signature.classes().tensor_name(), tensor_name_to_dtype, in BuildClassificationSignatureDef()
75 classification_signature.scores().tensor_name(), tensor_name_to_dtype, in BuildClassificationSignatureDef()
183 BuildClassificationSignatureDef(signature.classification_signature(), in ConvertDefaultSignatureToSignatureDef()
222 BuildClassificationSignatureDef(signature.classification_signature(), in ConvertNamedSignaturesToSignatureDef()
Dexporter.py95 def classification_signature(input_tensor, function
109 signature.classification_signature.input.tensor_name = input_tensor.name
111 signature.classification_signature.classes.tensor_name = classes_tensor.name
113 signature.classification_signature.scores.tensor_name = scores_tensor.name
Dbundle_shim_test.cc197 ClassificationSignature* classification_signature = in TEST() local
200 classification_signature->mutable_input()->set_tensor_name("foo-input"); in TEST()
201 classification_signature->mutable_classes()->set_tensor_name("foo-classes"); in TEST()
202 classification_signature->mutable_scores()->set_tensor_name("foo-scores"); in TEST()
Dsignature.cc91 *signature = signatures.default_signature().classification_signature(); in GetClassificationSignature()
111 *signature = it->second.classification_signature(); in GetNamedClassificationSignature()
Dexporter_test.py116 default_graph_signature=exporter.classification_signature(
167 default_signature.classification_signature.input.tensor_name, "v0:0")
Dmanifest.proto40 ClassificationSignature classification_signature = 2; field
Dsignature_test.cc217 EXPECT_EQ(signature.classification_signature().input().tensor_name(), "flow"); in TEST()
484 .classification_signature() in TEST()
DREADME.md90 ClassificationSignature classification_signature = 2;
100 signature = exporter.classification_signature(input_tensor=input,
/external/tensorflow/tensorflow/contrib/learn/python/learn/utils/
Dexport.py146 default_signature = exporter.classification_signature(
149 default_signature = exporter.classification_signature(
178 default_signature = exporter.classification_signature(
181 default_signature = exporter.classification_signature(