Lines Matching refs:dataset
20 #include "tensorflow/core/kernels/data/dataset.h"
31 class Dataset : public GraphDatasetBase {
33 explicit Dataset(OpKernelContext* ctx,
54 return "SparseTensorSliceDatasetOp::Dataset";
79 class Iterator : public DatasetIterator<Dataset<T>> {
82 : DatasetIterator<Dataset<T>>(params),
83 num_elements_(params.dataset->sparse_tensor_.shape()[0]),
84 dense_shape_(DT_INT64, {params.dataset->sparse_tensor_.dims() - 1}),
85 group_iterable_(params.dataset->sparse_tensor_.group({0})),
89 params.dataset->sparse_tensor_.shape()[i + 1];
104 const int rank = Iterator::dataset()->sparse_tensor_.dims();
216 // Create a new SparseTensorSliceDatasetOp::Dataset, insert it in
242 // appropriate order (e.g. by `tf.parse_example()` or a Dataset
258 *output = new Dataset<T>(ctx, sparse_tensor);
