| 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(); |
nothing calls this directly
no outgoing calls
no test coverage detected