Lines Matching defs:OpFoldResult
58 static OpFoldResult getDimValue(OpBuilder &builder, Location loc, Value v,
77 ArrayRef<OpFoldResult> offsets,
78 ArrayRef<OpFoldResult> sizes,
79 ArrayRef<OpFoldResult> strides) {
105 OpFoldResult linalg::createFoldedDimOp(OpBuilder &b, Location loc, Value source,
770 SmallVector<OpFoldResult> lowPads = srcPadOp.getMixedLowPad();
771 SmallVector<OpFoldResult> oldOffsets = insertOp.getMixedOffsets();
782 SmallVector<OpFoldResult, 4> newOffsets;
789 SmallVector<OpFoldResult, 4> newSizes;
920 OpFoldResult firstFillVal =
932 OpFoldResult fillVal =
1328 LogicalResult GenericOp::fold(FoldAdaptor, SmallVectorImpl<OpFoldResult> &) {
1957 SmallVectorImpl<OpFoldResult> &result) {
2027 SmallVector<OpFoldResult> dims;
2040 SmallVector<OpFoldResult> transposeResultShapes =
2650 ArrayRef<OpFoldResult> offsets,
2651 ArrayRef<OpFoldResult> sizes) {
2654 SmallVector<OpFoldResult> strides(rank, oneAttr);
2682 OpBuilder &builder, unsigned resultNumber, ArrayRef<OpFoldResult> offsets,
2683 ArrayRef<OpFoldResult> sizes, SmallVector<OpFoldResult> &resultOffsets,
2684 SmallVector<OpFoldResult> &resultSizes) {
2694 LogicalResult SoftmaxOp::fold(FoldAdaptor, SmallVectorImpl<OpFoldResult> &) {
2701 SmallVector<OpFoldResult> shapes;
2712 OpFoldResult ofr = createOrFoldDimOp(b, loc, getInput(), dim);
2885 SmallVector<OpFoldResult> dims = tensor::getMixedSizes(b, loc, input);
2974 OpBuilder &builder, unsigned resultNumber, ArrayRef<OpFoldResult> offsets,
2975 ArrayRef<OpFoldResult> sizes, SmallVector<OpFoldResult> &resultOffsets,
2976 SmallVector<OpFoldResult> &resultSizes) {
3005 OpBuilder &builder, ArrayRef<OpFoldResult> offsets,
3006 ArrayRef<OpFoldResult> sizes) {
3016 SmallVector<OpFoldResult> sliceOffsets, sliceSizes;
3023 SmallVector<OpFoldResult> filterStrides(filterRank, oneAttr);
3029 SmallVector<OpFoldResult> resultOffsets, resultSizes;
3035 SmallVector<OpFoldResult> outputStrides(outputRank, oneAttr);
3121 OpBuilder &builder, unsigned resultNumber, ArrayRef<OpFoldResult> offsets,
3122 ArrayRef<OpFoldResult> sizes, SmallVector<OpFoldResult> &resultOffsets,
3123 SmallVector<OpFoldResult> &resultSizes) {
3157 ArrayRef<OpFoldResult> offsets,
3158 ArrayRef<OpFoldResult> sizes) {
3188 SmallVector<OpFoldResult> sliceOffsets, sliceSizes;
3190 OpFoldResult offsetH = OpFoldResult(mappedOffsetH);
3191 OpFoldResult offsetW = OpFoldResult(mappedOffsetW);
3194 OpFoldResult sizeH =
3195 alphaH != 1 ? OpFoldResult(mappedSizeH) : OpFoldResult(oneAttr);
3196 OpFoldResult sizeW =
3197 alphaW != 1 ? OpFoldResult(mappedSizeW) : OpFoldResult(oneAttr);
3201 SmallVector<OpFoldResult> inputStrides(inputRank, oneAttr);
3206 SmallVector<OpFoldResult> resultOffsets, resultSizes;
3212 SmallVector<OpFoldResult> outputStrides(outputRank, oneAttr);
3298 OpBuilder &builder, unsigned resultNumber, ArrayRef<OpFoldResult> offsets,
3299 ArrayRef<OpFoldResult> sizes, SmallVector<OpFoldResult> &resultOffsets,
3300 SmallVector<OpFoldResult> &resultSizes) {
3326 OpFoldResult offsetH = OpFoldResult(mappedOffsetH);
3327 OpFoldResult offsetW = OpFoldResult(mappedOffsetW);
3328 OpFoldResult sizeH =
3329 valueH != 1 ? OpFoldResult(mappedSizeH) : OpFoldResult(oneAttr);
3330 OpFoldResult sizeW =
3331 valueW != 1 ? OpFoldResult(mappedSizeW) : OpFoldResult(oneAttr);
3347 OpBuilder &builder, ArrayRef<OpFoldResult> offsets,
3348 ArrayRef<OpFoldResult> sizes) {
3353 SmallVector<OpFoldResult> sliceOffsets, sliceSizes;
3369 SmallVector<OpFoldResult> sliceStrides(valueRank, oneAttr);
3374 SmallVector<OpFoldResult> resultOffsets, resultSizes;
3380 SmallVector<OpFoldResult> strides(outputRank, oneAttr);
3597 LogicalResult MatmulOp::fold(FoldAdaptor, SmallVectorImpl<OpFoldResult> &) {
3784 LogicalResult ContractOp::fold(FoldAdaptor, SmallVectorImpl<OpFoldResult> &) {