Searched defs:mode_mean (Results 1 – 1 of 1) sorted by relevance
84 int64_t stride, int mode_mean, const index_t *bag_size, in compute_grad_weight_bags()219 bool mode_mean, in embedding_backward_cuda_kernel()