Searched refs:classification_signature (Results 1 - 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/contrib/session_bundle/ |
H A D | bundle_shim.py | 97 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 D | bundle_shim_test.py | 180 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 D | bundle_shim.cc | 64 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 D | exporter.py | 95 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 D | bundle_shim_test.cc | 193 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 D | exporter_test.py | 116 default_graph_signature=exporter.classification_signature( 167 default_signature.classification_signature.input.tensor_name, "v0:0")
|
H A D | signature.cc | 91 *signature = signatures.default_signature().classification_signature(); 111 *signature = it->second.classification_signature();
|
H A D | signature_test.cc | 216 EXPECT_EQ(signature.classification_signature().input().tensor_name(), "flow"); 480 .classification_signature()
|
/external/tensorflow/tensorflow/contrib/learn/python/learn/utils/ |
H A D | export.py | 140 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