Home
last modified time | relevance | path

Searched refs:signature_def (Results 1 – 25 of 48) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/session_bundle/
Dbundle_shim_test.py50 signature_def = meta_graph_pb2.SignatureDef()
55 signature_def)
56 self.assertEqual(len(signature_def.inputs), 1)
57 self.assertEqual(len(signature_def.outputs), 0)
59 signature_def.inputs["foo-key"],
65 signature_def)
66 self.assertEqual(len(signature_def.inputs), 1)
67 self.assertEqual(len(signature_def.outputs), 0)
69 signature_def.inputs["foo-key"],
74 signature_def)
[all …]
Dbundle_shim.py36 def _add_input_to_signature_def(tensor_name, map_key, signature_def): argument
49 signature_def.inputs[map_key].CopyFrom(tensor_info)
52 def _add_output_to_signature_def(tensor_name, map_key, signature_def): argument
66 signature_def.outputs[map_key].CopyFrom(tensor_info)
84 signature_def = meta_graph_pb2.SignatureDef()
88 signature_def.method_name = signature_constants.REGRESS_METHOD_NAME
91 signature_def)
94 signature_def)
98 signature_def.method_name = signature_constants.CLASSIFY_METHOD_NAME
101 signature_def)
[all …]
Dbundle_shim.cc52 SignatureDef signature_def; in BuildRegressionSignatureDef() local
53 signature_def.set_method_name(kRegressMethodName); in BuildRegressionSignatureDef()
56 &signature_def); in BuildRegressionSignatureDef()
59 &signature_def); in BuildRegressionSignatureDef()
60 return signature_def; in BuildRegressionSignatureDef()
66 SignatureDef signature_def; in BuildClassificationSignatureDef() local
67 signature_def.set_method_name(kClassifyMethodName); in BuildClassificationSignatureDef()
70 kClassifyInputs, &signature_def); in BuildClassificationSignatureDef()
73 kClassifyOutputClasses, &signature_def); in BuildClassificationSignatureDef()
76 kClassifyOutputScores, &signature_def); in BuildClassificationSignatureDef()
[all …]
Dbundle_shim_test.cc77 const auto& signature_def_map = meta_graph_def.signature_def(); in LoadAndValidateSavedModelBundle()
109 auto iter = meta_graph_def.signature_def().find(key); in ValidateSignatureDef()
110 ASSERT_TRUE(iter != meta_graph_def.signature_def().end()); in ValidateSignatureDef()
116 SignatureDef signature_def; in TEST() local
125 &signature_def); in TEST()
126 EXPECT_EQ(1, signature_def.inputs_size()); in TEST()
127 EXPECT_EQ(tensor_name, signature_def.inputs().find(map_key)->second.name()); in TEST()
132 SignatureDef signature_def; in TEST() local
141 &signature_def); in TEST()
142 EXPECT_EQ(1, signature_def.outputs_size()); in TEST()
[all …]
Dbundle_shim.h41 const string& input_map_key, SignatureDef* signature_def);
49 const string& output_map_key, SignatureDef* signature_def);
/external/tensorflow/tensorflow/contrib/saved_model/cc/saved_model/
Dsignature_def_utils.cc54 bool IsValidPredictSignature(const SignatureDef& signature_def) { in IsValidPredictSignature() argument
55 if (signature_def.method_name() != kPredictMethodName) { in IsValidPredictSignature()
58 if (signature_def.inputs().empty()) { in IsValidPredictSignature()
61 if (signature_def.outputs().empty()) { in IsValidPredictSignature()
67 bool IsValidRegressionSignature(const SignatureDef& signature_def) { in IsValidRegressionSignature() argument
68 if (signature_def.method_name() != kRegressMethodName) { in IsValidRegressionSignature()
71 if (!VerifyTensorInfoForKeyInMap(signature_def.inputs(), kRegressInputs, in IsValidRegressionSignature()
75 if (!VerifyTensorInfoForKeyInMap(signature_def.outputs(), kRegressOutputs, in IsValidRegressionSignature()
82 bool IsValidClassificationSignature(const SignatureDef& signature_def) { in IsValidClassificationSignature() argument
83 if (signature_def.method_name() != kClassifyMethodName) { in IsValidClassificationSignature()
[all …]
Dsignature_def_utils_test.cc37 SignatureDef* signature_def) { in SetInputNameForKey() argument
38 (*signature_def->mutable_inputs())[key].set_name(name); in SetInputNameForKey()
42 SignatureDef* signature_def) { in SetOutputNameForKey() argument
43 (*signature_def->mutable_outputs())[key].set_name(name); in SetOutputNameForKey()
70 const SignatureDef* signature_def; in TEST_F() local
73 FindSignatureDefByKey(meta_graph_def, kSignatureKey, &signature_def)); in TEST_F()
74 EXPECT_EQ(kMethodName, signature_def->method_name()); in TEST_F()
77 FindSignatureDefByKey(meta_graph_def, "nonexistent", &signature_def) in TEST_F()
82 const SignatureDef signature_def = MakeSampleSignatureDef(); in TEST_F() local
85 TF_ASSERT_OK(FindInputTensorNameByKey(signature_def, kInput2Key, &name)); in TEST_F()
[all …]
Dsignature_def_utils.h37 const SignatureDef** signature_def);
43 Status FindInputTensorInfoByKey(const SignatureDef& signature_def,
51 Status FindOutputTensorInfoByKey(const SignatureDef& signature_def,
58 Status FindInputTensorNameByKey(const SignatureDef& signature_def,
64 Status FindOutputTensorNameByKey(const SignatureDef& signature_def,
68 bool IsValidSignature(const SignatureDef& signature_def);
/external/tensorflow/tensorflow/python/saved_model/
Dsignature_def_utils_impl.py52 signature_def = meta_graph_pb2.SignatureDef()
55 signature_def.inputs[item].CopyFrom(inputs[item])
58 signature_def.outputs[item].CopyFrom(outputs[item])
60 signature_def.method_name = method_name
61 return signature_def
105 signature_def = build_signature_def(
109 return signature_def
164 signature_def = build_signature_def(
168 return signature_def
205 signature_def = build_signature_def(
[all …]
Dsignature_def_utils_test.py75 signature_def = signature_def_utils_impl.build_signature_def(
77 self.assertEqual("foo-method-name", signature_def.method_name)
80 self.assertEqual(1, len(signature_def.inputs))
81 x_tensor_info_actual = signature_def.inputs["foo-input"]
88 self.assertEqual(1, len(signature_def.outputs))
89 y_tensor_info_actual = signature_def.outputs["foo-output"]
98 signature_def = signature_def_utils_impl.regression_signature_def(
102 signature_def.method_name)
105 self.assertEqual(1, len(signature_def.inputs))
107 signature_def.inputs[signature_constants.REGRESS_INPUTS])
[all …]
Dsimple_save_test.py87 signature_def_map = graph.signature_def
92 signature_def = signature_def_map[
95 signature_def.method_name)
97 self.assertEqual(1, len(signature_def.inputs))
98 self._check_tensor_info(signature_def.inputs["x"], var_x)
99 self.assertEqual(1, len(signature_def.outputs))
100 self._check_tensor_info(signature_def.outputs["y"], var_y)
Dload_v1_in_v2.py100 for signature_key, signature_def in meta_graph_def.signature_def.items():
101 input_names, input_specs = zip(*signature_def.inputs.items())
107 for name, out in signature_def.outputs.items()})
Dbuilder_impl.py149 meta_graph_def.signature_def[key].CopyFrom(signature_def_map[key])
198 signature_def = signature_def_map[signature_def_key]
199 inputs = signature_def.inputs
200 outputs = signature_def.outputs
/external/tensorflow/tensorflow/contrib/predictor/
Dsaved_model_predictor.py70 signature_def = metagraph_def.signature_def[signature_def_key]
75 signature_def = metagraph_def.signature_def[formatted_key]
80 signature_def_key, list(metagraph_def.signature_def), e))
83 return signature_def
87 signature_def,
92 signature_def_specified = signature_def is not None
115 signature_def=None, argument
147 signature_def_key, signature_def, input_names, output_names)
156 if signature_def is None:
157 signature_def = _get_signature_def(signature_def_key, export_dir, tags)
[all …]
Dpredictor_factories.py110 signature_def=None, argument
148 signature_def=signature_def,
Dcore_estimator_predictor.py71 signature_def = _get_signature_def(
79 feed_tensor_info = signature_def.inputs
82 fetch_tensor_info = signature_def.outputs
DREADME.md54 signature_def = tf.saved_model.signature_def_utils.build_signature_def(
61 signature_def=signature_def)
Dsaved_model_predictor_test.py105 signature_def = signature_def_utils.build_signature_def(
111 signature_def=signature_def)
/external/tensorflow/tensorflow/lite/python/
Dconvert_saved_model.py77 signature_def_map = meta_graph.signature_def
89 def get_inputs_outputs(signature_def): argument
98 inputs_tensor_info = signature_def.inputs
99 outputs_tensor_info = signature_def.outputs
254 signature_def = get_signature_def(meta_graph, signature_key)
255 inputs, outputs = get_inputs_outputs(signature_def)
/external/tensorflow/tensorflow/cc/tools/
Dfreeze_saved_model_test.cc41 SignatureDef signature_def; in BuildSignatureDef() local
43 (*signature_def.mutable_inputs())[input].set_name(input); in BuildSignatureDef()
46 (*signature_def.mutable_outputs())[output].set_name(output); in BuildSignatureDef()
48 return signature_def; in BuildSignatureDef()
52 void AddSignatureDefToSavedModelBundle(const SignatureDef& signature_def, in AddSignatureDefToSavedModelBundle() argument
56 (*meta_graph_def->mutable_signature_def())[key] = signature_def; in AddSignatureDefToSavedModelBundle()
91 SignatureDef signature_def = in AddGraphDefWithOutputsToSavedModelBundle() local
93 AddSignatureDefToSavedModelBundle(signature_def, "signature_def", in AddGraphDefWithOutputsToSavedModelBundle()
279 SignatureDef signature_def = in TEST_F() local
281 AddSignatureDefToSavedModelBundle(signature_def, "signature_def", in TEST_F()
Dfreeze_saved_model.cc54 for (auto& sigdef_elem : saved_model_bundle.meta_graph_def.signature_def()) { in GetSignatureDefsInputsAndOutputs()
55 const SignatureDef& signature_def = sigdef_elem.second; in GetSignatureDefsInputsAndOutputs() local
56 for (auto& input_elem : signature_def.inputs()) { in GetSignatureDefsInputsAndOutputs()
59 for (auto& output_elem : signature_def.outputs()) { in GetSignatureDefsInputsAndOutputs()
/external/tensorflow/tensorflow/python/tools/
Dsaved_model_cli.py98 return meta_graph_def.signature_def[signature_def_key].inputs
116 return meta_graph_def.signature_def[signature_def_key].outputs
155 meta_graph_def.signature_def[signature_def_key].method_name)
243 return meta_graph.signature_def
618 if args.signature_def is None:
621 _show_inputs_outputs(args.dir, args.tag_set, args.signature_def)
640 run_saved_model_with_feed_dict(args.dir, args.tag_set, args.signature_def,
/external/tensorflow/tensorflow/cc/saved_model/
Dloader_test.cc74 const auto signature_def_map = bundle.meta_graph_def.signature_def(); in CheckSavedModelBundle()
75 const auto signature_def = signature_def_map.at("regress_x_to_y"); in CheckSavedModelBundle() local
77 const string input_name = signature_def.inputs().at(kRegressInputs).name(); in CheckSavedModelBundle()
79 signature_def.outputs().at(kRegressOutputs).name(); in CheckSavedModelBundle()
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
Dsaved_model_utils.py108 predict_signature = signatures.signature_def[
153 filter_signature = signatures.signature_def[
203 filter_signature = signatures.signature_def[
/external/tensorflow/tensorflow/core/protobuf/
Dmeta_graph.proto82 // signature_def: Map from user supplied key for a signature to a single
84 map<string, SignatureDef> signature_def = 5; field
248 // might have the following signature_def map.
255 // signature_def {
278 // signature_def {

12