Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- IPluginV2* L2NormHelperPluginCreator::createPlugin(const char* name, const PluginFieldCollection* fc)
- {
- const PluginField* fields = fc->fields;
- for (int i = 0; i < fc->nbFields; ++i)
- {
- const char* attrName = fields[i].name;
- if (!strcmp(attrName, "op_type"))
- {
- ASSERT(fields[i].type == PluginFieldType::kINT32);
- mOpType = static_cast<int>(*(static_cast<const int*>(fields[i].data)));
- }
- if (!strcmp(attrName, "eps"))
- {
- ASSERT(fields[i].type == PluginFieldType::kFLOAT32);
- mEps = static_cast<float>(*(static_cast<const float*>(fields[i].data)));
- }
- }
- L2NormHelper* obj = new L2NormHelper(mOpType, mEps);
- obj->setPluginNamespace(mNamespace.c_str());
- return obj;
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement