xref: /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/embedding.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #include <torch/nn/options/embedding.h>
2 
3 namespace torch {
4 namespace nn {
EmbeddingOptions(int64_t num_embeddings,int64_t embedding_dim)5 EmbeddingOptions::EmbeddingOptions(
6     int64_t num_embeddings,
7     int64_t embedding_dim)
8     : num_embeddings_(num_embeddings), embedding_dim_(embedding_dim) {}
9 
EmbeddingBagOptions(int64_t num_embeddings,int64_t embedding_dim)10 EmbeddingBagOptions::EmbeddingBagOptions(
11     int64_t num_embeddings,
12     int64_t embedding_dim)
13     : num_embeddings_(num_embeddings), embedding_dim_(embedding_dim) {}
14 } // namespace nn
15 } // namespace torch
16