Home
last modified time | relevance | path

Searched refs:row_ndims (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddense_to_sparse_batch_dataset_op.cc167 const int row_ndims = row_shape.dims(); in GetNextInternal() local
171 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()