Method operator()

example/cpp/bert_model.cpp:242–248

    : m_(new Impl(filename, device_type, n_layers, n_heads)) {}

std::vector<float> BertModel::operator()(
    const std::vector<std::vector<int64_t>> &inputs,
    const std::vector<std::vector<int64_t>> &poistion_ids,
    const std::vector<std::vector<int64_t>> &segment_ids, PoolType pooling,
    bool use_pooler) const {
  return m_->operator()(inputs, poistion_ids, segment_ids, pooling, use_pooler);
}

BertModel::~BertModel() = default;