Searched refs:dense_shape_vec (Results 1 – 1 of 1) sorted by relevance
172 auto dense_shape_vec = dense_shape.vec<int64>(); in GetNextInternal() local175 dense_shape_vec(i + 1) = 0; in GetNextInternal()177 dense_shape_vec(i + 1) = row_shape.dim_size(i); in GetNextInternal()208 dense_shape_vec(j + 1) = in GetNextInternal()210 dense_shape_vec(j + 1)); in GetNextInternal()269 dense_shape_vec(0) = batch_elements.size(); in GetNextInternal()