Searched defs:indices_start (Results 1 – 4 of 4) sorted by relevance
123 Eigen::DSizes<Eigen::DenseIndex, 2> indices_start(0, 1); in Compute() local
445 Eigen::DSizes<Eigen::DenseIndex, 2> indices_start(0, 1); in Compute() local
113 int64_t indices_start = offsets[t * B + b]; in embedding_bag_nbits_rowwise_offsets_kernel() local
307 let indices_start = slice_range.start.add(F::Len::SIZE); in shift() localVariable