Searched defs:grad_row (Results 1 – 2 of 2) sorted by relevance
162 const int grad_row = ((int) indices[idx]) * stride; in embedding_backward_kernel() local
84 const int64_t grad_row = ((int64_t) indices[idx]) * stride + z * numel * stride; in indexing_backward_kernel() local149 int64_t grad_row = 0; in indexing_backward_kernel_stride_1() local207 int64_t grad_row = 0; in indexing_backward_kernel_small_stride() local247 const int64_t grad_row = ((int64_t) indices[idx]) * stride + z * numel * stride; in indexing_backward_kernel_quantized() local