Home
last modified time | relevance | path

Searched refs:paddedShape (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DTransforms.cpp217 SmallVector<int64_t> paddedShape(shape.begin(), shape.end()); in padOperandToSmallestStaticBoundingBox() local
230 paddedShape[shapeIdx++] = in padOperandToSmallestStaticBoundingBox()
241 paddedShape[shapeIdx++] = *upperBound; in padOperandToSmallestStaticBoundingBox()
248 paddedShape, getElementTypeOrSelf(opOperand->get())); in padOperandToSmallestStaticBoundingBox()
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalgNamed.cpp45 SmallVector<int64_t, 4> paddedShape; in applyPad() local
52 paddedShape.push_back(inputShape[i]); in applyPad()
54 paddedShape.push_back(inputShape[i] + highPad + lowPad); in applyPad()
61 return tensor::createPadScalarOp(RankedTensorType::get(paddedShape, inputETy), in applyPad()