Method EncoderReplica

src/models/language_model.cc:340–350

    EncoderReplica::EncoderReplica(const std::shared_ptr<const LanguageModel>& model,
                                   std::unique_ptr<layers::Encoder> encoder)
      : SequenceEncoderReplica(model)
      , _model(model)
      , _encoder(std::move(encoder))
      , _pooler_activation(model->get_enum_value<ops::ActivationType>("pooler_activation"))
      , _pooler_dense(layers::build_optional_layer<layers::Dense>(*model,
                                                                  "pooler_dense",
                                                                  &_pooler_activation))
    {
    }

    EncoderForwardOutput
    EncoderReplica::forward_impl(const StorageView& ids,