Searched defs:ddim (Results 1 – 1 of 1) sorted by relevance
130 int64_t ddim = src.size(1); in index_select_add() local201 int64_t ddim = src.size(1); in index_select_add() local379 int64_t ddim = src.size(1); in index_select_add() local518 int64_t ddim = src.size(1); in index_select_scale_add() local566 int64_t ddim = src.size(1); in index_select_scale_add() local755 int64_t ddim = src.size(1); in index_select_scale_add() local1576 int64_t ddim = grad.size(1); in _embedding_bag_dense_backward_cpu_sum_mean() local