MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / AddBuiltin

Method AddBuiltin

tensorflow/lite/mutable_op_resolver.cc:32–43  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

30}
31
32void MutableOpResolver::AddBuiltin(tflite::BuiltinOperator op,
33 const TfLiteRegistration* registration,
34 int min_version, int max_version) {
35 for (int version = min_version; version <= max_version; ++version) {
36 TfLiteRegistration new_registration = *registration;
37 new_registration.custom_name = nullptr;
38 new_registration.builtin_code = op;
39 new_registration.version = version;
40 auto op_key = std::make_pair(op, version);
41 builtins_[op_key] = new_registration;
42 }
43}
44
45void MutableOpResolver::AddCustom(const char* name,
46 const TfLiteRegistration* registration,

Callers 1

TESTFunction · 0.45

Calls

no outgoing calls

Tested by 1

TESTFunction · 0.36