Function NodeSetParamGeneric

core/lib/node.cpp:199–239

}

int NodeSetParamGeneric(void* node, const char* param_name, const char* type_name, const void* param_val, int size)
{
    Node* real_node = ( Node* )node;

    Operator* op = real_node->GetOp();

    if(op->SetParamItem(param_name, type_name, param_val))
        return 0;

    /* check custom attr */
    if(!real_node->ExistAttr(ATTR_CUSTOM_ATTR))
    {
        set_tengine_errno(ENOENT);
        return -1;
    }

    node_custom_attr_map_t* attr_map = any_cast<node_custom_attr_map_t>(&real_node->GetAttr(ATTR_CUSTOM_ATTR));

    if(attr_map->count(param_name) == 0)
    {
        set_tengine_errno(ENOENT);
        return -1;
    }

    CustomNodeAttr* attr_entry = &attr_map->at(param_name);

    if((size != attr_entry->attr_size) ||
       (type_name && attr_entry->type_name && strcmp(type_name, attr_entry->type_name)))
    {
        set_tengine_errno(EINVAL);
        return -1;
    }

    attr_entry->mem.resize(size);

    void* data = attr_entry->mem.data();

    memcpy(data, param_val, size);

    return 0;
}

}    // namespace TEngine