Searched defs:user_emb (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | deep_wide_pt_bench.cc | 16 auto user_emb = torch::randn({batch_size, 1, embedding_size}); in BM_deep_wide_base() local
31 auto user_emb = torch::randn({batch_size, 1, embedding_size}); in BM_deep_wide_fast() local
45 auto user_emb = torch::randn({batch_size, 1, embedding_size}); in BM_deep_wide_jit_graph_executor() local
63 auto user_emb = torch::randn({batch_size, 1, embedding_size}); in BM_deep_wide_jit_profiling_executor() local
82 auto user_emb = torch::randn({batch_size, 1, embedding_size}); in BM_deep_wide_static() local
105 auto user_emb = torch::randn({batch_size, 1, embedding_size}); in BM_deep_wide_static_threaded() local
|
| H A D | test_static_module.cc | 494 auto user_emb = torch::randn({batch_size, 1, embedding_size}); in TEST() local
522 auto user_emb = torch::randn({batch_size, 1, embedding_size}); in TEST() local
563 auto user_emb = torch::randn({batch_size, 1, embedding_size}); in TEST() local
644 auto user_emb = torch::randn({batch_size, 1, embedding_size}); in TEST() local
757 auto user_emb = torch::randn({batch_size, 1, embedding_size}); in TEST() local
780 auto user_emb = torch::randn({batch_size, 1, embedding_size}); in TEST() local
886 auto user_emb = torch::randn({batch_size, 1, embedding_size}); in TEST() local
|