Method createPlugin

plugin/embLayerNormPlugin/embLayerNormPlugin.cpp:532–629

}

IPluginV2* EmbLayerNormPluginDynamicCreator::createPlugin(char const* name, PluginFieldCollection const* fc) noexcept
{
    try
    {
        BERT_DEBUG_MSG("EmbLayerNormPluginDynamic createPlugin.");

        bool output_fp16 = false;
        bool useFullMask = false;
        Weights beta{};     // required attribute - validateRequiredAttributesExist() will verify existence
        Weights gamma{};    // required attribute - validateRequiredAttributesExist() will verify existence
        Weights word_emb{}; // required attribute - validateRequiredAttributesExist() will verify existence
        Weights pos_emb{};  // required attribute - validateRequiredAttributesExist() will verify existence
        Weights tok_emb{};  // required attribute - validateRequiredAttributesExist() will verify existence
        int32_t mhaTypeId = 0;
        std::set<std::string> const requiredAttributes{
            "bert_embeddings_layernorm_beta",
            "bert_embeddings_layernorm_gamma",
            "bert_embeddings_word_embeddings",
            "bert_embeddings_token_type_embeddings",
            "bert_embeddings_position_embeddings",
        };
        plugin::validateRequiredAttributesExist(requiredAttributes, fc);

        for (int32_t i = 0; i < fc->nbFields; i++)
        {
            std::string field_name(fc->fields[i].name);
            if (field_name.compare("bert_embeddings_layernorm_beta") == 0)
            {
                BERT_DEBUG_MSG("Building bert_embeddings_layernorm_beta...");
                beta.values = fc->fields[i].data;
                beta.count = fc->fields[i].length;
                beta.type = fieldTypeToDataType(fc->fields[i].type);
            }

            if (field_name.compare("bert_embeddings_layernorm_gamma") == 0)
            {
                BERT_DEBUG_MSG("Building bert_embeddings_layernorm_gamma...");
                gamma.values = fc->fields[i].data;
                gamma.count = fc->fields[i].length;
                gamma.type = fieldTypeToDataType(fc->fields[i].type);
            }

            if (field_name.compare("bert_embeddings_word_embeddings") == 0)
            {
                BERT_DEBUG_MSG("Building bert_embeddings_word_embeddings...");
                word_emb.values = fc->fields[i].data;
                word_emb.count = fc->fields[i].length;
                word_emb.type = fieldTypeToDataType(fc->fields[i].type);
            }

            if (field_name.compare("bert_embeddings_token_type_embeddings") == 0)
            {
                BERT_DEBUG_MSG("Building bert_embeddings_token_type_embeddings...");
                tok_emb.values = fc->fields[i].data;
                tok_emb.count = fc->fields[i].length;
                tok_emb.type = fieldTypeToDataType(fc->fields[i].type);
            }