Searched defs:weight_row (Results 1 – 3 of 3) sorted by relevance
62 auto weight_row = weights + c; in generic_pack() local
161 const int weight_row = ((int) input[idx]) * stride; in embedding_backward_kernel() local
83 const int64_t weight_row = ((int64_t) sorted_indices[idx]) * stride + z * stride_before; in indexing_backward_kernel() local148 const int64_t weight_row = crnt_sorted_idx * stride + z * stride_before; in indexing_backward_kernel_stride_1() local206 const int64_t weight_row = crnt_sorted_idx * stride + z * stride_before; in indexing_backward_kernel_small_stride() local246 const int64_t weight_row = ((int64_t) sorted_indices[idx]) * stride + z * stride_before; in indexing_backward_kernel_quantized() local