Home
last modified time | relevance | path

Searched refs:batch_idx_end (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dgather_functor.h63 SliceIndex batch_idx_end = static_cast<SliceIndex>(end / indices_size); in HandleCopies() local
66 while ((batch_idx < batch_idx_end) || in HandleCopies()
67 (batch_idx == batch_idx_end && indices_idx < indices_idx_end)) { in HandleCopies()
70 if ((batch_idx == batch_idx_end && i_next < indices_idx_end) || in HandleCopies()
76 } else if (b_next <= batch_idx_end) { in HandleCopies()