Lines Matching defs:tensor

172     return builder.create<tensor::FromElementsOp>(loc, rtp, elem);
550 sparse_tensor::genToMemref(OpBuilder &builder, Location loc, Value tensor) {
551   auto tTp = llvm::cast<TensorType>(tensor.getType());
553   return builder.create<bufferization::ToMemrefOp>(loc, mTp, tensor)
558                                                Value tensor, Dimension dim) {
559   auto enc = getSparseTensorEncoding(tensor.getType());
564   return builder.create<ToSliceOffsetOp>(loc, tensor, APInt(64, dim));
568                                                Value tensor, Dimension dim) {
569   auto enc = getSparseTensorEncoding(tensor.getType());
574   return builder.create<ToSliceStrideOp>(loc, tensor, APInt(64, dim));
578                                SparseTensorType stt, Value tensor,
599                      {tensor, dimShapesBuffer, valTp}, EmitCInterface::On)
682       llvm::report_fatal_error("unsupported dim2lvl in sparse tensor type");
725       llvm::report_fatal_error("unsupported lvl2dim in sparse tensor type");