Lines Matching refs:elemTp

225   Type elemTp = stp.getElementType();  in newParams()  local
228 params.push_back(constantPrimaryTypeEncoding(builder, loc, elemTp)); in newParams()
259 static void genDelCOOCall(OpBuilder &builder, Operation *op, Type elemTp, in genDelCOOCall() argument
261 SmallString<21> name{"delSparseTensorCOO", primaryTypeFunctionSuffix(elemTp)}; in genDelCOOCall()
285 Type elemTp = elemPtr.getType().cast<ShapedType>().getElementType(); in genGetNextCall() local
286 SmallString<10> name{"getNext", primaryTypeFunctionSuffix(elemTp)}; in genGetNextCall()
330 Type elemTp = tensorTp.getElementType(); in allocDenseTensor() local
332 auto memTp = MemRefType::get(shape, elemTp); in allocDenseTensor()
339 Value zero = constantZero(builder, loc, elemTp); in allocDenseTensor()
469 Type elemTp = srcTp.getElementType(); in genSparse2SparseReshape() local
470 assert(elemTp == dstTp.getElementType() && in genSparse2SparseReshape()
492 Value elemPtr = genAllocaScalar(rewriter, loc, elemTp); in genSparse2SparseReshape()
505 genAddEltCall(rewriter, op, elemTp, coo, elemPtr, dstIdx, dstPerm); in genSparse2SparseReshape()
512 genDelCOOCall(rewriter, op, elemTp, coo); in genSparse2SparseReshape()
513 genDelCOOCall(rewriter, op, elemTp, iter); in genSparse2SparseReshape()
738 Type elemTp = dstTensorTp.getElementType(); in matchAndRewrite() local
754 Value elemPtr = genAllocaScalar(rewriter, loc, elemTp); in matchAndRewrite()
771 genDelCOOCall(rewriter, op, elemTp, iter); in matchAndRewrite()
966 Type elemTp = op.getTensor().getType().cast<ShapedType>().getElementType(); in matchAndRewrite() local
967 SmallString<12> name{"lexInsert", primaryTypeFunctionSuffix(elemTp)}; in matchAndRewrite()
1030 Type elemTp = op.getTensor().getType().cast<ShapedType>().getElementType(); in matchAndRewrite() local
1031 SmallString<12> name{"expInsert", primaryTypeFunctionSuffix(elemTp)}; in matchAndRewrite()