Lines Matching refs:reader

123       SparseTensorReader &reader = *static_cast<SparseTensorReader *>(ptr);    \
124       return static_cast<void *>(reader.readSparseTensor<P, C, V>(             \
375   auto *reader = SparseTensorReader::create(filename, dimRank, dimShape, valTp);  local
376   return static_cast<void *>(reader);
382   SparseTensorReader &reader = *static_cast<SparseTensorReader *>(p);  in _mlir_ciface_getSparseTensorReaderDimSizes()  local
383   auto *dimSizes = const_cast<uint64_t *>(reader.getDimSizes());  in _mlir_ciface_getSparseTensorReaderDimSizes()
384   aliasIntoMemref(reader.getRank(), dimSizes, *out);  in _mlir_ciface_getSparseTensorReaderDimSizes()
393     auto &reader = *static_cast<SparseTensorReader *>(p);                      \
398     const uint64_t dimRank = reader.getRank();                                 \
403     assert(cSize >= lvlRank * reader.getNSE());                                \
404     assert(vSize >= reader.getNSE());                                          \
412     return reader.readToBuffers<C, V>(lvlRank, dim2lvl, lvl2dim,               \