Searched refs:indices_i (Results 1 – 3 of 3) sorted by relevance
38 Index indices_i = 0; in GatherOpKernel() local41 indices_i = i / slice_size; in GatherOpKernel()42 slice_i = i - indices_i * slice_size; in GatherOpKernel()48 indices_i = batch_indices_i - batch_i * indices_size; in GatherOpKernel()54 Index gather_i = ldg(indices + indices_i); in GatherOpKernel()
80 int indices_i = i / update_block;82 int param_first_index = indices[indices_i];101 int indices_i = i / update_block;102 int param_first_index = indices[indices_i];
39 const auto indices_i = indices + IXDIM * loc; in GatherSliceOpKernel() local44 const Index index_j = ldg(indices_i + j); in GatherSliceOpKernel()