MCPcopy Create free account
hub / github.com/OpenNMT/CTranslate2 / ParallelEmbeddings

Method ParallelEmbeddings

src/layers/common.cc:84–94  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

82
83
84 ParallelEmbeddings::ParallelEmbeddings(const models::Model& model,
85 const std::string& scope,
86 const EmbeddingsMerge merge)
87 : _merge(merge)
88 {
89 auto single_layer = build_optional_layer<Embeddings>(model, scope);
90 if (single_layer)
91 _layers.emplace_back(std::move(single_layer));
92 else
93 _layers = build_layers_list<const Embeddings>(model, scope);
94 }
95
96 DataType ParallelEmbeddings::output_type() const {
97 return _layers[0]->output_type();

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected