Searched refs:slice_i (Results 1 – 3 of 3) sorted by relevance
39 Index slice_i = 0; in GatherOpKernel() local42 slice_i = i - indices_i * slice_size; in GatherOpKernel()50 slice_i = i - batch_indices_i * slice_size; in GatherOpKernel()66 (batch_i * gather_dim_size + gather_i) * slice_size + slice_i; in GatherOpKernel()
227 slice_i = Lambda(235 inputs.append(slice_i)
458 int64 slice_i = 0;464 result.values_[slice_i++] = values_[i];