|
@@ -181,12 +181,19 @@ void TBEPluginManager::GetCustomOpPath(std::string &customop_path) { |
|
|
void TBEPluginManager::LoadCustomOpLib() { |
|
|
void TBEPluginManager::LoadCustomOpLib() { |
|
|
LoadPluginSo(options_); |
|
|
LoadPluginSo(options_); |
|
|
|
|
|
|
|
|
|
|
|
std::string fmk_type = std::to_string(domi::TENSORFLOW); |
|
|
|
|
|
auto it = options_.find(ge::FRAMEWORK_TYPE); |
|
|
|
|
|
if (it != options_.end()) { |
|
|
|
|
|
fmk_type = it->second; |
|
|
|
|
|
} |
|
|
std::vector<OpRegistrationData> registration_datas = domi::OpRegistry::Instance()->registrationDatas; |
|
|
std::vector<OpRegistrationData> registration_datas = domi::OpRegistry::Instance()->registrationDatas; |
|
|
GELOGI("The size of registration_datas is: %zu", registration_datas.size()); |
|
|
GELOGI("The size of registration_datas is: %zu", registration_datas.size()); |
|
|
for (OpRegistrationData reg_data : registration_datas) { |
|
|
for (OpRegistrationData reg_data : registration_datas) { |
|
|
GELOGD("Begin to register optype: %s, imply_type: %s", reg_data.GetOmOptype().c_str(), |
|
|
|
|
|
TypeUtils::ImplyTypeToSerialString(reg_data.GetImplyType()).c_str()); |
|
|
|
|
|
domi::OpRegistry::Instance()->Register(reg_data); |
|
|
|
|
|
|
|
|
if (std::to_string(reg_data.GetFrameworkType()) == fmk_type) { |
|
|
|
|
|
GELOGD("Begin to register optype: %s, imply_type: %s", reg_data.GetOmOptype().c_str(), |
|
|
|
|
|
TypeUtils::ImplyTypeToSerialString(reg_data.GetImplyType()).c_str()); |
|
|
|
|
|
(void)domi::OpRegistry::Instance()->Register(reg_data); |
|
|
|
|
|
} |
|
|
} |
|
|
} |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|