Home
last modified time | relevance | path

Searched refs:SignatureDef (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/lite/tools/signature/
Dsignature_def_util_test.cc33 using tensorflow::SignatureDef;
49 SignatureDef GetTestSignatureDef() { in GetTestSignatureDef()
50 auto signature_def = SignatureDef(); in GetTestSignatureDef()
65 SignatureDef expected_signature_def = GetTestSignatureDef(); in TEST_F()
67 const std::map<string, SignatureDef> expected_signature_def_map = { in TEST_F()
73 std::map<string, SignatureDef> test_signature_def_map; in TEST_F()
76 SignatureDef test_signature_def = in TEST_F()
85 std::map<string, SignatureDef> expected_signature_def_map = { in TEST_F()
91 std::map<string, SignatureDef> test_signature_def_map; in TEST_F()
94 SignatureDef test_signature_def = in TEST_F()
[all …]
Dsignature_def_util.h45 const std::map<std::string, tensorflow::SignatureDef>& signature_def_map,
61 std::map<std::string, tensorflow::SignatureDef>* signature_def_map);
Dsignature_def_util_wrapper_pybind11.cc32 std::map<std::string, tensorflow::SignatureDef> signature_def_map; in WrappedSetSignatureDefMap()
34 tensorflow::SignatureDef signature_def; in WrappedSetSignatureDefMap()
56 std::map<std::string, tensorflow::SignatureDef> signature_def_map; in WrappedGetSignatureDefMap()
Dsignature_def_util.cc33 using SignatureDefMap = std::map<std::string, tensorflow::SignatureDef>;
139 tensorflow::SignatureDef signature_def; in GetSignatureDefMap()
Dsignature_def_utils.py72 signature_def = meta_graph_pb2.SignatureDef()
/external/tensorflow/tensorflow/cc/saved_model/
Dloader.h42 virtual const protobuf::Map<string, SignatureDef>& GetSignatures() const = 0;
61 const protobuf::Map<string, SignatureDef>& GetSignatures() const override { in GetSignatures()
78 protobuf::Map<string, SignatureDef> signatures) in SavedModelBundleLite()
90 const protobuf::Map<string, SignatureDef>& GetSignatures() const override { in GetSignatures()
96 protobuf::Map<string, SignatureDef> signatures_;
/external/tensorflow/tensorflow/cc/tools/
Dfreeze_saved_model_test.cc39 SignatureDef BuildSignatureDef(const std::unordered_set<string>& inputs, in BuildSignatureDef()
41 SignatureDef signature_def; in BuildSignatureDef()
52 void AddSignatureDefToSavedModelBundle(const SignatureDef& signature_def, in AddSignatureDefToSavedModelBundle()
91 SignatureDef signature_def = in AddGraphDefWithOutputsToSavedModelBundle()
279 SignatureDef signature_def = in TEST_F()
296 SignatureDef signature_def_0 = BuildSignatureDef({"input0:0"}, {"output0:0"}); in TEST_F()
297 SignatureDef signature_def_1 = BuildSignatureDef({"input1:0"}, {"output1:0"}); in TEST_F()
432 SignatureDef signature_def; in TEST_F()
459 SignatureDef signature_def; in TEST_F()
Dfreeze_saved_model.cc59 const SignatureDef& signature_def = sigdef_elem.second; in GetSignatureDefsInputsAndOutputs()
/external/tensorflow/tensorflow/core/protobuf/
Dmeta_graph.proto33 // SignatureDef
88 // SignatureDef.
89 map<string, SignatureDef> signature_def = 5;
260 // SignatureDef defines the signature of a computation supported by a TensorFlow
317 message SignatureDef { message
323 // SignatureDef as supporting a particular method. This enables producers and
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-meta-graph-def.-signature-def-entry.pbtxt16 type_name: ".tensorflow.SignatureDef"
Dtensorflow.-meta-graph-def.pbtxt158 type_name: ".tensorflow.SignatureDef"
/external/tensorflow/tensorflow/python/saved_model/model_utils/
Dexport_output_test.py78 expected_signature_def = meta_graph_pb2.SignatureDef()
111 expected_signature_def = meta_graph_pb2.SignatureDef()
147 expected_signature_def = meta_graph_pb2.SignatureDef()
188 expected_signature_def = meta_graph_pb2.SignatureDef()
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder_test.cc225 SignatureDef serving_signature; in TEST_F()
235 SignatureDef serving_signature2; in TEST_F()
267 SignatureDef serving_signature; in TEST_F()
294 SignatureDef serving_signature; in TEST_F()
/external/tensorflow/tensorflow/go/
Dsignature.go93 func signatureDefFromProto(pb *corepb.SignatureDef) Signature { argument
Dsaved_model.go94 func generateSignatures(pb map[string]*corepb.SignatureDef) map[string]Signature { argument
Dsignature_test.go29 got := signatureDefFromProto(&corepb.SignatureDef{
/external/tensorflow/tensorflow/lite/
Dinterpreter_test_util.h55 Interpreter::SignatureDef signature; in BuildSignature()
Dinterpreter.h664 struct SignatureDef { struct
718 void SetSignatureDef(std::vector<SignatureDef> signature_defs) { in SetSignatureDef()
771 std::vector<SignatureDef> signature_defs_;
Dinterpreter_builder.h101 const flatbuffers::Vector<flatbuffers::Offset<SignatureDef>>*
Dinterpreter_builder.cc491 const flatbuffers::Vector<flatbuffers::Offset<SignatureDef>>* in ParseSignatureDefs()
497 std::vector<Interpreter::SignatureDef> signature_defs; in ParseSignatureDefs()
/external/llvm-project/llvm/cmake/modules/
DTensorFlowCompile.cmake2 # ${model} directory, looking for the ${tag_set} tag set, and the SignatureDef
/external/tensorflow/tensorflow/python/saved_model/
Dsignature_def_utils_impl.py52 signature_def = meta_graph_pb2.SignatureDef()
/external/tensorflow/tensorflow/lite/schema/
Dschema.fbs1137 // This corresponds to SignatureDef in Tensorflow SavedModel.
1138 // The SignatureDef will be part of the SavedModel provided for conversion.
1139 table SignatureDef {
1149 // Key value which was in the Tensorflow SavedModel SignatureDef map.
1183 signature_defs:[SignatureDef];
/external/tensorflow/tensorflow/python/tools/
Dsaved_model_aot_compile.py198 new_signature_def = meta_graph_pb2.SignatureDef()
/external/tensorflow/tensorflow/lite/python/
Dutil.py269 signature = _meta_graph_pb2.SignatureDef()

12