Searched refs:batch_indices_i (Results 1 – 1 of 1) sorted by relevance
44 Index batch_indices_i = i / slice_size; in GatherOpKernel() local46 batch_i = batch_indices_i / indices_size; in GatherOpKernel()48 indices_i = batch_indices_i - batch_i * indices_size; in GatherOpKernel()50 slice_i = i - batch_indices_i * slice_size; in GatherOpKernel()