Searched refs:row_ndims (Results 1 – 1 of 1) sorted by relevance
167 const int row_ndims = row_shape.dims(); in GetNextInternal() local171 Tensor dense_shape(ctx->allocator({}), DT_INT64, {row_ndims + 1}); in GetNextInternal()173 for (size_t i = 0; i < row_ndims; ++i) { in GetNextInternal()198 if (batch_element_tuple[0].shape().dims() != row_ndims) { in GetNextInternal()205 for (int j = 0; j < row_ndims; ++j) { in GetNextInternal()232 {total_elements, row_ndims + 1}); in GetNextInternal()246 gtl::InlinedVector<int64, 4> strides(row_ndims); in GetNextInternal()248 strides[row_ndims - 1] = 1; in GetNextInternal()