Searched defs:stride_warped (Results 1 – 1 of 1) sorted by relevance
88 const int64_t stride_warped) { in compute_grad_weight_bags()133 const int64_t stride_warped) { in compute_grad_weight()166 const int64_t stride_warped) { in sum_and_scatter()303 const int stride_warped = ceil_div(stride, warp_size)*warp_size; in embedding_backward_cuda_kernel() local