Searched refs:indices_values (Results 1 – 4 of 4) sorted by relevance
14 int batch = is_1d ? 0 : indices_values(idx, 0);
13 const auto indices_values = indices.matrix<int64>();
26 int batch = is_1d ? 0 : indices_values(idx, 0);
205 const auto indices_values = indices.matrix<int64>(); in Compute() local214 int batch = is_1d ? 0 : indices_values(idx, 0); in Compute()