Home
last modified time | relevance | path

Searched refs:signature_def (Results 1 – 25 of 32) 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.cc73 const auto& signature_def_map = meta_graph_def.signature_def(); in LoadAndValidateSavedModelBundle()
105 auto iter = meta_graph_def.signature_def().find(key); in ValidateSignatureDef()
106 ASSERT_TRUE(iter != meta_graph_def.signature_def().end()); in ValidateSignatureDef()
112 SignatureDef signature_def; in TEST() local
121 &signature_def); in TEST()
122 EXPECT_EQ(1, signature_def.inputs_size()); in TEST()
123 EXPECT_EQ(tensor_name, signature_def.inputs().find(map_key)->second.name()); in TEST()
128 SignatureDef signature_def; in TEST() local
137 &signature_def); in TEST()
138 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/python/saved_model/
Dsignature_def_utils_impl.py46 signature_def = meta_graph_pb2.SignatureDef()
49 signature_def.inputs[item].CopyFrom(inputs[item])
52 signature_def.outputs[item].CopyFrom(outputs[item])
54 signature_def.method_name = method_name
55 return signature_def
93 signature_def = build_signature_def(
97 return signature_def
146 signature_def = build_signature_def(
150 return signature_def
181 signature_def = build_signature_def(
[all …]
Dsignature_def_utils_test.py72 signature_def = signature_def_utils_impl.build_signature_def(
74 self.assertEqual("foo-method-name", signature_def.method_name)
77 self.assertEqual(1, len(signature_def.inputs))
78 x_tensor_info_actual = signature_def.inputs["foo-input"]
85 self.assertEqual(1, len(signature_def.outputs))
86 y_tensor_info_actual = signature_def.outputs["foo-output"]
94 signature_def = signature_def_utils_impl.regression_signature_def(
98 signature_def.method_name)
101 self.assertEqual(1, len(signature_def.inputs))
103 signature_def.inputs[signature_constants.REGRESS_INPUTS])
[all …]
Dsimple_save_test.py85 signature_def_map = graph.signature_def
90 signature_def = signature_def_map[
93 signature_def.method_name)
95 self.assertEqual(1, len(signature_def.inputs))
96 self._check_tensor_info(signature_def.inputs["x"], var_x)
97 self.assertEqual(1, len(signature_def.outputs))
98 self._check_tensor_info(signature_def.outputs["y"], var_y)
Dbuilder_impl.py188 meta_graph_def.signature_def[key].CopyFrom(signature_def_map[key])
230 signature_def = signature_def_map[signature_def_key]
231 inputs = signature_def.inputs
232 outputs = signature_def.outputs
/external/tensorflow/tensorflow/contrib/saved_model/python/saved_model/
Dsignature_def_utils_test.py37 meta_graph_def.signature_def[key].CopyFrom(signature_def_map[key])
106 signature_def = signature_def_contrib_utils.get_signature_def_by_key(
111 signature_def.method_name)
114 self.assertEqual(1, len(signature_def.inputs))
115 self._check_tensor_info(signature_def.inputs,
119 self.assertEqual(1, len(signature_def.outputs))
120 self._check_tensor_info(signature_def.outputs,
136 signature_def = signature_def_contrib_utils.get_signature_def_by_key(
141 signature_def.method_name)
144 self.assertEqual(1, len(signature_def.inputs))
[all …]
Dsignature_def_utils.py39 if signature_def_key not in meta_graph_def.signature_def:
42 return meta_graph_def.signature_def[signature_def_key]
/external/tensorflow/tensorflow/contrib/saved_model/cc/saved_model/
Dsignature_def_utils.cc40 const SignatureDef** signature_def) { in FindSignatureDefByKey() argument
41 return FindInProtobufMap("SignatureDef", meta_graph_def.signature_def(), in FindSignatureDefByKey()
42 signature_def_key, signature_def); in FindSignatureDefByKey()
45 Status FindInputTensorInfoByKey(const SignatureDef& signature_def, in FindInputTensorInfoByKey() argument
48 return FindInProtobufMap("input TensorInfo", signature_def.inputs(), in FindInputTensorInfoByKey()
52 Status FindOutputTensorInfoByKey(const SignatureDef& signature_def, in FindOutputTensorInfoByKey() argument
55 return FindInProtobufMap("output TensorInfo", signature_def.outputs(), in FindOutputTensorInfoByKey()
59 Status FindInputTensorNameByKey(const SignatureDef& signature_def, in FindInputTensorNameByKey() argument
63 FindInputTensorInfoByKey(signature_def, tensor_info_key, &tensor_info)); in FindInputTensorNameByKey()
68 Status FindOutputTensorNameByKey(const SignatureDef& signature_def, in FindOutputTensorNameByKey() argument
[all …]
Dsignature_def_utils_test.cc59 const SignatureDef* signature_def; in TEST_F() local
62 FindSignatureDefByKey(meta_graph_def, kSignatureKey, &signature_def)); in TEST_F()
63 EXPECT_EQ(kMethodName, signature_def->method_name()); in TEST_F()
66 FindSignatureDefByKey(meta_graph_def, "nonexistent", &signature_def) in TEST_F()
71 const SignatureDef signature_def = MakeSampleSignatureDef(); in TEST_F() local
74 TF_ASSERT_OK(FindInputTensorNameByKey(signature_def, kInput2Key, &name)); in TEST_F()
78 FindInputTensorNameByKey(signature_def, "nonexistent", &name).ok()); in TEST_F()
82 const SignatureDef signature_def = MakeSampleSignatureDef(); in TEST_F() local
85 TF_ASSERT_OK(FindOutputTensorNameByKey(signature_def, kOutput2Key, &name)); in TEST_F()
89 FindOutputTensorNameByKey(signature_def, "nonexistent", &name).ok()); in TEST_F()
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,
/external/tensorflow/tensorflow/contrib/predictor/
Dsaved_model_predictor.py71 signature_def = signature_def_utils.get_signature_def_by_key(
78 signature_def = signature_def_utils.get_signature_def_by_key(
87 signature_def_key, list(metagraph_def.signature_def), e))
88 return signature_def
92 signature_def,
97 signature_def_specified = signature_def is not None
120 signature_def=None, argument
150 signature_def_key, signature_def, input_names, output_names)
159 if signature_def is None:
160 signature_def = _get_signature_def(signature_def_key, export_dir, tags)
[all …]
Dpredictor_factories.py101 signature_def=None, argument
129 signature_def=signature_def,
Dcore_estimator_predictor.py69 signature_def = _get_signature_def(
76 feed_tensor_info = signature_def.inputs
79 fetch_tensor_info = signature_def.outputs
Dsaved_model_predictor_test.py105 signature_def = signature_def_utils.build_signature_def(
111 signature_def=signature_def)
DREADME.md54 signature_def = tf.saved_model.signature_def_utils.build_signature_def(
61 signature_def=signature_def)
/external/tensorflow/tensorflow/cc/tools/
Dfreeze_saved_model_test.cc40 SignatureDef signature_def; in BuildSignatureDef() local
42 (*signature_def.mutable_inputs())[input].set_name(input); in BuildSignatureDef()
45 (*signature_def.mutable_outputs())[output].set_name(output); in BuildSignatureDef()
47 return signature_def; in BuildSignatureDef()
51 void AddSignatureDefToSavedModelBundle(const SignatureDef& signature_def, in AddSignatureDefToSavedModelBundle() argument
55 (*meta_graph_def->mutable_signature_def())[key] = signature_def; in AddSignatureDefToSavedModelBundle()
90 SignatureDef signature_def = in AddGraphDefWithOutputsToSavedModelBundle() local
92 AddSignatureDefToSavedModelBundle(signature_def, "signature_def", in AddGraphDefWithOutputsToSavedModelBundle()
124 SignatureDef signature_def = in TEST_F() local
126 AddSignatureDefToSavedModelBundle(signature_def, "signature_def", in TEST_F()
Dfreeze_saved_model.cc53 for (auto& sigdef_elem : saved_model_bundle.meta_graph_def.signature_def()) { in GetSignatureDefsInputsAndOutputs()
54 const SignatureDef& signature_def = sigdef_elem.second; in GetSignatureDefsInputsAndOutputs() local
55 for (auto& input_elem : signature_def.inputs()) { in GetSignatureDefsInputsAndOutputs()
58 for (auto& output_elem : signature_def.outputs()) { in GetSignatureDefsInputsAndOutputs()
/external/tensorflow/tensorflow/cc/saved_model/
Dloader_test.cc71 const auto signature_def_map = bundle.meta_graph_def.signature_def(); in CheckSavedModelBundle()
72 const auto signature_def = signature_def_map.at("regress_x_to_y"); in CheckSavedModelBundle() local
74 const string input_name = signature_def.inputs().at(kRegressInputs).name(); in CheckSavedModelBundle()
76 signature_def.outputs().at(kRegressOutputs).name(); in CheckSavedModelBundle()
/external/tensorflow/tensorflow/python/tools/
Dsaved_model_cli.py148 meta_graph_def.signature_def[signature_def_key].method_name)
230 return meta_graph.signature_def
573 if args.signature_def is None:
576 _show_inputs_outputs(args.dir, args.tag_set, args.signature_def)
595 run_saved_model_with_feed_dict(args.dir, args.tag_set, args.signature_def,
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
Dsaved_model_utils.py105 predict_signature = signatures.signature_def[
151 filter_signature = signatures.signature_def[
/external/tensorflow/tensorflow/core/protobuf/
Dmeta_graph.proto81 // signature_def: Map from user supplied key for a signature to a single
83 map<string, SignatureDef> signature_def = 5; field
244 // might have the following signature_def map.
251 // signature_def {
274 // signature_def {

12