Searched refs:classification_signature (Results 1 - 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/contrib/session_bundle/
H A 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,
H A 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
[all...]
H A Dbundle_shim.cc64 const ClassificationSignature& classification_signature,
69 classification_signature.input().tensor_name(), tensor_name_to_dtype,
72 classification_signature.classes().tensor_name(), tensor_name_to_dtype,
75 classification_signature.scores().tensor_name(), tensor_name_to_dtype,
183 BuildClassificationSignatureDef(signature.classification_signature(),
222 BuildClassificationSignatureDef(signature.classification_signature(),
63 BuildClassificationSignatureDef( const ClassificationSignature& classification_signature, const std::unordered_map<string, DataType>& tensor_name_to_dtype) argument
H A 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
H A Dbundle_shim_test.cc193 ClassificationSignature* classification_signature = local
196 classification_signature->mutable_input()->set_tensor_name("foo-input");
197 classification_signature->mutable_classes()->set_tensor_name("foo-classes");
198 classification_signature->mutable_scores()->set_tensor_name("foo-scores");
H A Dexporter_test.py116 default_graph_signature=exporter.classification_signature(
167 default_signature.classification_signature.input.tensor_name, "v0:0")
H A Dsignature.cc91 *signature = signatures.default_signature().classification_signature();
111 *signature = it->second.classification_signature();
H A Dsignature_test.cc216 EXPECT_EQ(signature.classification_signature().input().tensor_name(), "flow");
480 .classification_signature()
/external/tensorflow/tensorflow/contrib/learn/python/learn/utils/
H A Dexport.py140 default_signature = exporter.classification_signature(
143 default_signature = exporter.classification_signature(
172 default_signature = exporter.classification_signature(
175 default_signature = exporter.classification_signature(

Completed in 140 milliseconds