Searched defs:dp_unflatten (Results 1 – 1 of 1) sorted by relevance
26 auto dp_unflatten = torch::bmm(ad_emb_packed, user_emb_t); in forward() local60 auto dp_unflatten = at::cpu::bmm(ad_emb_packed, user_emb_t); in forward() local