createPlugin.cpp

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;
}