/external/tensorflow/tensorflow/contrib/saved_model/python/saved_model/ |
H A D | signature_def_utils.py | 39 if signature_def_key not in meta_graph_def.signature_def: 42 return meta_graph_def.signature_def[signature_def_key]
|
H A D | signature_def_utils_test.py | 37 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 D | signature_def_utils.h | 31 // 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 D | signature_def_utils.cc | 40 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 D | signature_def_utils_test.cc | 59 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 D | bundle_shim_test.py | 50 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 D | bundle_shim.py | 36 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 D | bundle_shim.cc | 52 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 D | bundle_shim.h | 41 const string& input_map_key, SignatureDef* signature_def); 49 const string& output_map_key, SignatureDef* signature_def);
|
H A D | bundle_shim_test.cc | 73 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 D | signature_def_utils_impl.py | 46 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 D | signature_def_utils_test.py | 72 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 D | simple_save_test.py | 85 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 D | builder_impl.py | 188 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 D | saved_model_predictor.py | 71 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 D | core_estimator_predictor.py | 69 signature_def = _get_signature_def( 76 feed_tensor_info = signature_def.inputs 79 fetch_tensor_info = signature_def.outputs
|
H A D | predictor_factories.py | 101 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 D | saved_model_predictor_test.py | 105 signature_def = signature_def_utils.build_signature_def( 111 signature_def=signature_def)
|
/external/tensorflow/tensorflow/cc/tools/ |
H A D | freeze_saved_model_test.cc | 40 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 D | freeze_saved_model.cc | 53 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 D | saved_model_utils.py | 105 predict_signature = signatures.signature_def[ 151 filter_signature = signatures.signature_def[
|
/external/tensorflow/tensorflow/cc/saved_model/ |
H A D | loader_test.cc | 71 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 D | saved_model_cli.py | 148 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 D | meta_graph_transform_test.py | 556 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 D | meta_graph_transform.py | 566 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:
|