HomeSort by relevance Sort by last modified time
    Searched refs:SignatureDef (Results 1 - 14 of 14) sorted by null

  /external/tensorflow/tensorflow/contrib/saved_model/cc/saved_model/
signature_def_utils.h 33 // SignatureDef* points into meta_graph_def and may be invalidated by changes
37 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,
67 // Determine whether a SignatureDef can be served by TensorFlow Serving.
68 bool IsValidSignature(const SignatureDef& signature_def);
signature_def_utils.cc 54 bool IsValidPredictSignature(const SignatureDef& signature_def) {
67 bool IsValidRegressionSignature(const SignatureDef& signature_def) {
82 bool IsValidClassificationSignature(const SignatureDef& signature_def) {
115 const SignatureDef** signature_def) {
116 return FindInProtobufMap("SignatureDef", meta_graph_def.signature_def(),
120 Status FindInputTensorInfoByKey(const SignatureDef& signature_def,
127 Status FindOutputTensorInfoByKey(const SignatureDef& signature_def,
134 Status FindInputTensorNameByKey(const SignatureDef& signature_def,
143 Status FindOutputTensorNameByKey(const SignatureDef& signature_def,
152 bool IsValidSignature(const SignatureDef& signature_def)
    [all...]
signature_def_utils_test.cc 37 SignatureDef* signature_def) {
42 SignatureDef* signature_def) {
46 SignatureDef MakeSampleSignatureDef() {
47 SignatureDef result;
70 const SignatureDef* signature_def;
82 const SignatureDef signature_def = MakeSampleSignatureDef();
93 const SignatureDef signature_def = MakeSampleSignatureDef();
123 SignatureDef signature_def_;
  /external/tensorflow/tensorflow/contrib/session_bundle/
bundle_shim.h 41 const string& input_map_key, SignatureDef* signature_def);
49 const string& output_map_key, SignatureDef* signature_def);
bundle_shim.cc 49 SignatureDef BuildRegressionSignatureDef(
52 SignatureDef signature_def;
63 SignatureDef BuildClassificationSignatureDef(
66 SignatureDef signature_def;
101 SignatureDef signature_def;
186 LOG(WARNING) << "Default signature up-conversion to SignatureDef is only "
208 // If a Predict SignatureDef was successfully constructed, skip the entries
226 << "Named signature up-conversion to SignatureDef is only supported "
240 // Helper functions to populate SignatureDef fields.
242 // Adds an entry to the `inputs` map of the supplied SignatureDef
    [all...]
bundle_shim_test.py 50 signature_def = meta_graph_pb2.SignatureDef()
51 signature_def_compare = meta_graph_pb2.SignatureDef()
97 signature_def = meta_graph_pb2.SignatureDef()
98 signature_def_compare = meta_graph_pb2.SignatureDef()
bundle_shim.py 42 signature_def: object of type meta_graph_pb2.SignatureDef()
58 signature_def: object of type meta_graph_pb2.SignatureDef()
70 """Convert default signature to object of type SignatureDef.
76 object of type SignatureDef which contains a converted version of default
80 to SignatureDef.
84 signature_def = meta_graph_pb2.SignatureDef()
117 """Convert named signatures to object of type SignatureDef.
123 object of type SignatureDef which contains a converted version of named
130 signature_def = meta_graph_pb2.SignatureDef()
155 """Produce default and named upconverted SignatureDef objects from Signatures
    [all...]
bundle_shim_test.cc 81 SignatureDef regression_signature_def = regression_entry->second;
102 // Helper function to validate that the SignatureDef found in the MetaGraphDef
106 tensorflow::SignatureDef expected_signature;
116 SignatureDef signature_def;
132 SignatureDef signature_def;
354 // Checks the Predict SignatureDef created when the named signatures have
483 // SignatureDef that corresponds to the kDefaultServingSignatureDefKey.
492 // Verify that the `Predict` SignatureDef is created under a different key.
534 const SignatureDef& signature_def = entry.second;
  /external/tensorflow/tensorflow/cc/tools/
freeze_saved_model_test.cc 38 // Builds a SignatureDef with the provided `inputs` and `outputs`.
39 SignatureDef BuildSignatureDef(const std::unordered_set<string>& inputs,
41 SignatureDef signature_def;
52 void AddSignatureDefToSavedModelBundle(const SignatureDef& signature_def,
86 // Adds `graph_def` and `outputs` as the GraphDef and SignatureDef in
91 SignatureDef signature_def =
120 // in the SignatureDef. The resulting graph shouldn't be frozen, but
167 // SignatureDef. The variables should be frozen.
216 // the outputs in the SignatureDef. The resulting graph should only freeze
275 // SignatureDef
    [all...]
freeze_saved_model.cc 55 const SignatureDef& signature_def = sigdef_elem.second;
  /external/tensorflow/tensorflow/core/grappler/
grappler_item_builder_test.cc 225 SignatureDef serving_signature;
235 SignatureDef serving_signature2;
267 SignatureDef serving_signature;
294 SignatureDef serving_signature;
  /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/
saved_model_export_utils_test.py 88 expected_signature_def = meta_graph_pb2.SignatureDef()
119 expected_signature_def = meta_graph_pb2.SignatureDef()
159 expected_signature_def = meta_graph_pb2.SignatureDef()
205 expected_signature_def = meta_graph_pb2.SignatureDef()
248 expected_signature_def = meta_graph_pb2.SignatureDef()
292 expected_signature_def = meta_graph_pb2.SignatureDef()
333 expected_signature_def = meta_graph_pb2.SignatureDef()
    [all...]
  /external/tensorflow/tensorflow/python/saved_model/model_utils/
export_output_test.py 77 expected_signature_def = meta_graph_pb2.SignatureDef()
110 expected_signature_def = meta_graph_pb2.SignatureDef()
146 expected_signature_def = meta_graph_pb2.SignatureDef()
187 expected_signature_def = meta_graph_pb2.SignatureDef()
  /external/tensorflow/tensorflow/python/saved_model/
signature_def_utils_impl.py 15 """SignatureDef utility functions implementation."""
40 """Utility function to build a SignatureDef protocol buffer.
43 inputs: Inputs of the SignatureDef defined as a proto map of string to
45 outputs: Outputs of the SignatureDef defined as a proto map of string to
47 method_name: Method name of the SignatureDef as a string.
50 A SignatureDef protocol buffer constructed based on the supplied arguments.
52 signature_def = meta_graph_pb2.SignatureDef()
237 method_name: Method name of the SignatureDef as a string.
276 """Determine whether a SignatureDef can be served by TensorFlow Serving."""
285 """Determine whether the argument is a servable 'predict' SignatureDef.""
    [all...]

Completed in 878 milliseconds