Home
last modified time | relevance | path

Searched refs:getI64ArrayAttr (Results 1 – 25 of 31) sorted by relevance

12

/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaDecomposeTransposeConv.cpp132 rewriter.getI64ArrayAttr(convPad), rewriter.getI64ArrayAttr(stride), in matchAndRewrite()
137 rewriter.getI64ArrayAttr(convPad), rewriter.getI64ArrayAttr(stride), in matchAndRewrite()
138 rewriter.getI64ArrayAttr({1, 1})); in matchAndRewrite()
227 rewriter.getI64ArrayAttr(weightReshapeDims0)); in matchAndRewrite()
244 rewriter.getI64ArrayAttr(weightReshapeDims1)); in matchAndRewrite()
296 /*stride=*/rewriter.getI64ArrayAttr({1, 1}), in matchAndRewrite()
297 /*dilation=*/rewriter.getI64ArrayAttr({1, 1}), in matchAndRewrite()
305 /*stride=*/rewriter.getI64ArrayAttr({1, 1}), in matchAndRewrite()
322 rewriter.getI64ArrayAttr(convReshapeDims0)); in matchAndRewrite()
338 rewriter.getI64ArrayAttr(convReshapeDims1)); in matchAndRewrite()
[all …]
H A DTosaDecomposeDepthwise.cpp69 rewriter.getI64ArrayAttr(revisedInputShape)) in matchAndRewrite()
81 rewriter.getI64ArrayAttr(revisedWeightShape)) in matchAndRewrite()
104 rewriter.getI64ArrayAttr(outputShape)); in matchAndRewrite()
H A DTosaDecomposeConv2D.cpp64 rewriter.getI64ArrayAttr(revisedInputShape)) in matchAndRewrite()
76 rewriter.getI64ArrayAttr(revisedWeightShape)) in matchAndRewrite()
105 rewriter.getI64ArrayAttr(outputShape)); in matchAndRewrite()
H A DTosaMakeBroadcastable.cpp126 rewriter.getI64ArrayAttr(reshapeOutputShape)); in reshapeLowerToHigher()
/llvm-project-15.0.7/mlir/lib/Conversion/LLVMCommon/
H A DStructBuilder.cpp29 builder.getI64ArrayAttr(pos)); in extractPtr()
35 builder.getI64ArrayAttr(pos)); in setPtr()
H A DMemRefBuilder.cpp109 builder.getI64ArrayAttr(kOffsetPosInMemRefDescriptor)); in offset()
117 builder.getI64ArrayAttr(kOffsetPosInMemRefDescriptor)); in setOffset()
131 builder.getI64ArrayAttr({kSizePosInMemRefDescriptor, pos})); in size()
145 builder.getI64ArrayAttr({kSizePosInMemRefDescriptor})); in size()
161 builder.getI64ArrayAttr({kSizePosInMemRefDescriptor, pos})); in setSize()
174 builder.getI64ArrayAttr({kStridePosInMemRefDescriptor, pos})); in stride()
182 builder.getI64ArrayAttr({kStridePosInMemRefDescriptor, pos})); in setStride()
H A DVectorPattern.cpp76 auto position = builder.getI64ArrayAttr(coords); in nDVectorIterate()
H A DPattern.cpp340 op->getLoc(), type, newOp->getResult(0), rewriter.getI64ArrayAttr(i))); in oneToOneRewrite()
/llvm-project-15.0.7/mlir/lib/Conversion/NVGPUToNVVM/
H A DNVGPUToNVVM.cpp90 rewriter.getI64ArrayAttr(i)); in convertIntrinsicResult()
110 rewriter.getI64ArrayAttr(i * 2)); in convertIntrinsicResult()
113 rewriter.getI64ArrayAttr(i * 2 + 1)); in convertIntrinsicResult()
127 rewriter.getI64ArrayAttr(el.index())); in convertIntrinsicResult()
156 loc, arrayTy.getElementType(), operand, rewriter.getI64ArrayAttr(i)); in unpackOperandVector()
244 rewriter.getI64ArrayAttr(i)) in matchAndRewrite()
249 loc, finalResultType, result, casted, rewriter.getI64ArrayAttr(i)); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToTensor/
H A DTosaToTensor.cpp54 ValueRange({}), starts, sizes, rewriter.getI64ArrayAttr(strides)); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransforms.cpp94 auto posAttr = rewriter.getI64ArrayAttr(pos); in reshapeLoad()
104 auto posAttr = rewriter.getI64ArrayAttr(d); in reshapeLoad()
123 auto posAttr = rewriter.getI64ArrayAttr(pos); in reshapeStore()
131 auto posAttr = rewriter.getI64ArrayAttr(d); in reshapeStore()
565 auto pos = rewriter.getI64ArrayAttr(d); in matchAndRewrite()
771 auto pos = rewriter.getI64ArrayAttr(d); in matchAndRewrite()
821 auto pos = rewriter.getI64ArrayAttr(d); in matchAndRewrite()
1296 rewriter.getI64ArrayAttr(invOrder))); in matchAndRewrite()
2360 newOffsets = rewriter.getI64ArrayAttr(offsets); in matchAndRewrite()
2370 newSizes = rewriter.getI64ArrayAttr(sizes); in matchAndRewrite()
[all …]
H A DVectorInsertExtractStridedSliceRewritePatterns.cpp229 rewriter.getI64ArrayAttr(offsets)); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp1065 dynamicStrides, b.getI64ArrayAttr(staticOffsets), in build()
1066 b.getI64ArrayAttr(staticSizes), b.getI64ArrayAttr(staticStrides)); in build()
1472 dynamicStrides, b.getI64ArrayAttr(staticOffsets), in build()
1473 b.getI64ArrayAttr(staticSizes), b.getI64ArrayAttr(staticStrides)); in build()
1900 build(b, result, resultType, source, low, high, b.getI64ArrayAttr(staticLow), in build()
1901 b.getI64ArrayAttr(staticHigh), nofold ? b.getUnitAttr() : UnitAttr()); in build()
1935 b.getI64ArrayAttr(staticLow), b.getI64ArrayAttr(staticHigh), in build()
2264 dynamicStrides, b.getI64ArrayAttr(staticOffsets), in build()
2265 b.getI64ArrayAttr(staticSizes), b.getI64ArrayAttr(staticStrides)); in build()
/llvm-project-15.0.7/mlir/lib/Interfaces/
H A DViewLikeInterface.cpp141 integers = parser.getBuilder().getI64ArrayAttr(attrVals); in parseOperandsOrIntegersImpl()
241 return {b.getI64ArrayAttr(staticValues), dynamicValues}; in decomposeMixedImpl()
/llvm-project-15.0.7/mlir/lib/Dialect/Utils/
H A DReshapeOpsUtils.cpp164 return b.getI64ArrayAttr(indices).cast<Attribute>(); in getReassociationIndicesAttribute()
/llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/
H A DVectorToGPU.cpp564 builder.getI64ArrayAttr(i)); in createNonLdMatrixLoads()
588 op.getLoc(), el, result, builder.getI64ArrayAttr({i, innerIdx})); in createNonLdMatrixLoads()
691 op.getLoc(), opC.getType(), opA, opB, opC, b.getI64ArrayAttr({m, n, k})); in convertContractOpToMmaSync()
/llvm-project-15.0.7/mlir/lib/Conversion/FuncToLLVM/
H A DFuncToLLVM.cpp528 rewriter.getI64ArrayAttr(i))); in matchAndRewrite()
647 rewriter.getI64ArrayAttr(i)); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/
H A DMemRefOps.cpp1482 dynamicStrides, b.getI64ArrayAttr(staticOffsets), in build()
1483 b.getI64ArrayAttr(staticSizes), b.getI64ArrayAttr(staticStrides)); in build()
2221 dynamicStrides, b.getI64ArrayAttr(staticOffsets), in build()
2222 b.getI64ArrayAttr(staticSizes), b.getI64ArrayAttr(staticStrides)); in build()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp328 return builder.getI64ArrayAttr(values); in getVectorSubscriptAttr()
344 builder.getI64ArrayAttr(reductionDims)); in build()
502 rewriter.getI64ArrayAttr(0)); in matchAndRewrite()
1095 b.getI64ArrayAttr(globalPosition)); in foldExtractOpFromExtractChain()
1270 b.getI64ArrayAttr( in tryToFoldExtractOpInPlace()
1344 b.getI64ArrayAttr(extractPos)); in foldExtractFromBroadcast()
1402 b.getI64ArrayAttr(newPosition)); in foldExtractFromShapeCast()
1444 b.getI64ArrayAttr(extractedPos)); in foldExtractFromExtractStrided()
1501 b.getI64ArrayAttr(offsetDiffs)); in foldExtractStridedOpFromInsertChain()
2585 b.getI64ArrayAttr(offsetDiffs)); in foldExtractStridedOpFromInsertChain()
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/
H A DLinalgOps.td109 shape, $_builder.getI64ArrayAttr(staticShape));
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/
H A DOneShotModuleBufferize.cpp141 op->setAttr(kEquivalentArgsAttr, b.getI64ArrayAttr(equivBbArgs)); in annotateEquivalentReturnBbArg()
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DBuilders.h143 ArrayAttr getI64ArrayAttr(ArrayRef<int64_t> values);
/llvm-project-15.0.7/mlir/lib/IR/
H A DBuilders.cpp220 ArrayAttr Builder::getI64ArrayAttr(ArrayRef<int64_t> values) { in getI64ArrayAttr() function in Builder
/llvm-project-15.0.7/mlir/lib/Conversion/VectorToLLVM/
H A DConvertVectorToLLVM.cpp62 rewriter.getI64ArrayAttr(pos)); in insertOne()
78 rewriter.getI64ArrayAttr(pos)); in extractOne()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/
H A DAffineOps.cpp3310 result.addAttribute(getStepsAttrStrName(), builder.getI64ArrayAttr(steps)); in build()
3428 setStepsAttr(getBodyBuilder().getI64ArrayAttr(newSteps)); in setSteps()
3755 builder.getI64ArrayAttr(steps)); in parse()
3774 builder.getI64ArrayAttr(steps)); in parse()

12