Class CodebookLookupModule

include/engine/framework/modules/lookup_modules.h:69–83

};

class CodebookLookupModule {
public:
    explicit CodebookLookupModule(IndexedEmbeddingConfig config);

    const IndexedEmbeddingConfig & config() const noexcept;
    const core::ModuleSchema & schema() const noexcept;
    core::TensorValue build(
        core::ModuleBuildContext & ctx,
        const core::TensorValue & indices,
        const core::TensorValue & weight) const;
    static const core::ModuleSchema & static_schema() noexcept;

private:
    IndexedEmbeddingConfig config_;
};

}  // namespace engine::modules