D | Sparsification.cpp | 447 static void genBuffers(Merger &merger, CodeGen &codegen, in genBuffers() argument 475 dynShape, genIntType(rewriter, codegen.options.ptrType)); in genBuffers() 477 dynShape, genIntType(rewriter, codegen.options.indType)); in genBuffers() 478 codegen.pointers[t][i] = rewriter.create<AllocaOp>(loc, ptrTp, unknown); in genBuffers() 479 codegen.indices[t][i] = rewriter.create<AllocaOp>(loc, indTp, unknown); in genBuffers() 487 up = codegen.sizes[i]; in genBuffers() 497 codegen.sizes[i] = codegen.highs[t][i] = up; in genBuffers() 502 codegen.buffers[t] = rewriter.create<AllocaOp>(loc, denseTp, args); in genBuffers() 506 codegen.buffers[t] = rewriter.create<AllocaOp>(loc, sparseTp, unknown); in genBuffers() 512 static Value genTensorLoad(Merger &merger, CodeGen &codegen, in genTensorLoad() argument [all …]
|