Searched refs:SliceIndex (Results 1 – 2 of 2) sorted by relevance
37 template <typename T, typename Index, typename SliceIndex,38 SliceIndex static_slice_elems>39 SliceIndex HandleCopies(OpKernelContext* ctx, in HandleCopies()42 SliceIndex slice_elems, in HandleCopies()44 const SliceIndex indices_size = static_cast<SliceIndex>(indices.dimension(0)); in HandleCopies()45 const SliceIndex batch_size = static_cast<SliceIndex>(params.dimension(0)); in HandleCopies()59 SliceIndex result = -1; in HandleCopies()61 SliceIndex batch_idx = static_cast<SliceIndex>(start / indices_size); in HandleCopies()62 SliceIndex indices_idx = static_cast<SliceIndex>(start % indices_size); in HandleCopies()63 SliceIndex batch_idx_end = static_cast<SliceIndex>(end / indices_size); in HandleCopies()[all …]
37 SliceIndex interface { interface263 _ SliceIndex = sliceIndex{}