Searched refs:signature_def (Results 1 - 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/saved_model/python/saved_model/
H A 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]
H A 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
[all...]
/external/tensorflow/tensorflow/contrib/saved_model/cc/saved_model/
H A Dsignature_def_utils.h31 // Finds the entry in meta_graph_def.signature_def with the given key, or
32 // returns NotFound and leaves *signature_def unchanged. NOTE: The output
37 const SignatureDef** signature_def);
39 // Finds the entry in signature_def.inputs with the given key, or returns
41 // points into signature_def and may be invalidated by changes to that protocol
43 Status FindInputTensorInfoByKey(const SignatureDef& signature_def,
47 // Finds the entry in signature_def.outputs with the given key, or returns
49 // points into signature_def and may be invalidated by changes to that protocol
51 Status FindOutputTensorInfoByKey(const SignatureDef& signature_def,
55 // Finds the entry in signature_def
[all...]
H A Dsignature_def_utils.cc40 const SignatureDef** signature_def) {
41 return FindInProtobufMap("SignatureDef", meta_graph_def.signature_def(),
42 signature_def_key, signature_def);
45 Status FindInputTensorInfoByKey(const SignatureDef& signature_def, argument
48 return FindInProtobufMap("input TensorInfo", signature_def.inputs(),
52 Status FindOutputTensorInfoByKey(const SignatureDef& signature_def, argument
55 return FindInProtobufMap("output TensorInfo", signature_def.outputs(),
59 Status FindInputTensorNameByKey(const SignatureDef& signature_def, argument
63 FindInputTensorInfoByKey(signature_def, tensor_info_key, &tensor_info));
68 Status FindOutputTensorNameByKey(const SignatureDef& signature_def, argument
38 FindSignatureDefByKey(const MetaGraphDef& meta_graph_def, const string& signature_def_key, const SignatureDef** signature_def) argument
[all...]
H A Dsignature_def_utils_test.cc59 const SignatureDef* signature_def; local
62 FindSignatureDefByKey(meta_graph_def, kSignatureKey, &signature_def));
63 EXPECT_EQ(kMethodName, signature_def->method_name());
66 FindSignatureDefByKey(meta_graph_def, "nonexistent", &signature_def)
71 const SignatureDef signature_def = MakeSampleSignatureDef(); local
74 TF_ASSERT_OK(FindInputTensorNameByKey(signature_def, kInput2Key, &name));
78 FindInputTensorNameByKey(signature_def, "nonexistent", &name).ok());
82 const SignatureDef signature_def = MakeSampleSignatureDef(); local
85 TF_ASSERT_OK(FindOutputTensorNameByKey(signature_def, kOutput2Key, &name));
89 FindOutputTensorNameByKey(signature_def, "nonexisten
[all...]
/external/tensorflow/tensorflow/contrib/session_bundle/
H A 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...]
H A Dbundle_shim.py36 def _add_input_to_signature_def(tensor_name, map_key, signature_def):
37 """Add input tensor to signature_def.
40 tensor_name: string name of tensor to add to signature_def inputs
41 map_key: string key to key into signature_def inputs map
42 signature_def: object of type meta_graph_pb2.SignatureDef()
45 adds a TensorInfo with tensor_name to signature_def inputs map keyed with
49 signature_def.inputs[map_key].CopyFrom(tensor_info)
52 def _add_output_to_signature_def(tensor_name, map_key, signature_def):
53 """Add output tensor to signature_def.
56 tensor_name: string name of tensor to add to signature_def output
[all...]
H A Dbundle_shim.cc52 SignatureDef signature_def; local
53 signature_def.set_method_name(kRegressMethodName);
56 &signature_def);
59 &signature_def);
60 return signature_def;
66 SignatureDef signature_def; local
67 signature_def.set_method_name(kClassifyMethodName);
70 kClassifyInputs, &signature_def);
73 kClassifyOutputClasses, &signature_def);
76 kClassifyOutputScores, &signature_def);
101 SignatureDef signature_def; local
243 AddInputToSignatureDef( const string& tensor_name, const std::unordered_map<string, DataType>& tensor_name_to_dtype, const string& input_key, SignatureDef* signature_def) argument
273 AddOutputToSignatureDef( const string& tensor_name, const std::unordered_map<string, DataType>& tensor_name_to_dtype, const string& output_key, SignatureDef* signature_def) argument
[all...]
H A Dbundle_shim.h41 const string& input_map_key, SignatureDef* signature_def);
49 const string& output_map_key, SignatureDef* signature_def);
H A Dbundle_shim_test.cc73 const auto& signature_def_map = meta_graph_def.signature_def();
105 auto iter = meta_graph_def.signature_def().find(key);
106 ASSERT_TRUE(iter != meta_graph_def.signature_def().end());
112 SignatureDef signature_def; local
121 &signature_def);
122 EXPECT_EQ(1, signature_def.inputs_size());
123 EXPECT_EQ(tensor_name, signature_def.inputs().find(map_key)->second.name());
128 SignatureDef signature_def; local
137 &signature_def);
138 EXPECT_EQ(1, signature_def
529 const SignatureDef& signature_def = entry.second; local
[all...]
/external/tensorflow/tensorflow/python/saved_model/
H A 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
71 A regression-flavored signature_def.
93 signature_def = build_signature_def(
97 return signature_def
115 A classification-flavored signature_def.
146 signature_def
[all...]
H A 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
[all...]
H A 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)
H A 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/predictor/
H A 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
109 'You must specify at most one of signature_def_key OR signature_def OR'
120 signature_def=None,
131 `signature_def_key` and `signature_def` should be specified.
132 signature_def
[all...]
H A Dcore_estimator_predictor.py69 signature_def = _get_signature_def(
76 feed_tensor_info = signature_def.inputs
79 fetch_tensor_info = signature_def.outputs
H A Dpredictor_factories.py101 signature_def=None,
110 `signature_def_key` and `signature_def`
111 signature_def: A `SignatureDef` proto specifying the inputs and outputs
112 for prediction. Only one of `signature_def_key` and `signature_def`
123 ValueError: More than one of `signature_def_key` and `signature_def` is
129 signature_def=signature_def,
H A Dsaved_model_predictor_test.py105 signature_def = signature_def_utils.build_signature_def(
111 signature_def=signature_def)
/external/tensorflow/tensorflow/cc/tools/
H A Dfreeze_saved_model_test.cc40 SignatureDef signature_def; local
42 (*signature_def.mutable_inputs())[input].set_name(input);
45 (*signature_def.mutable_outputs())[output].set_name(output);
47 return signature_def;
50 // Adds `signature_def` to `saved_model_bundle` under `key`.
51 void AddSignatureDefToSavedModelBundle(const SignatureDef& signature_def, argument
55 (*meta_graph_def->mutable_signature_def())[key] = signature_def;
90 SignatureDef signature_def = local
92 AddSignatureDefToSavedModelBundle(signature_def, "signature_def",
124 SignatureDef signature_def = local
[all...]
H A Dfreeze_saved_model.cc53 for (auto& sigdef_elem : saved_model_bundle.meta_graph_def.signature_def()) {
54 const SignatureDef& signature_def = sigdef_elem.second; local
55 for (auto& input_elem : signature_def.inputs()) {
58 for (auto& output_elem : signature_def.outputs()) {
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
H A Dsaved_model_utils.py105 predict_signature = signatures.signature_def[
151 filter_signature = signatures.signature_def[
/external/tensorflow/tensorflow/cc/saved_model/
H A Dloader_test.cc71 const auto signature_def_map = bundle.meta_graph_def.signature_def();
72 const auto signature_def = signature_def_map.at("regress_x_to_y"); local
74 const string input_name = signature_def.inputs().at(kRegressInputs).name();
76 signature_def.outputs().at(kRegressOutputs).name();
/external/tensorflow/tensorflow/python/tools/
H A 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,
628 '--signature_def serving_default\n\n'
650 '--signature_def',
662 '--signature_def serving_default '
682 '--signature_def',
/external/tensorflow/tensorflow/contrib/meta_graph_transform/
H A Dmeta_graph_transform_test.py556 base_sig_keep = base_meta_graph_def.signature_def[signature_name_keep]
561 base_sig_remove = base_meta_graph_def.signature_def[signature_name_remove]
576 self.assertTrue(signature_name_keep in meta_graph_def.signature_def)
577 sig_keep = meta_graph_def.signature_def[signature_name_keep]
580 self.assertFalse(signature_name_remove in meta_graph_def.signature_def)
H A Dmeta_graph_transform.py566 base_signature = base_meta_graph_def.signature_def[signature_name]
573 meta_graph_def.signature_def[signature_name].CopyFrom(base_signature)
763 for signature_name in base_meta_graph_def.signature_def:

Completed in 512 milliseconds

12