Searched refs:outputTensor (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | HoistPadding.cpp | 504 Value outputTensor = b.create<tensor::ExtractSliceOp>( in hoistPaddingOnTensors() local
507 makeTransposeOp(b, loc, paddedTensor, outputTensor, transposeVector)); in hoistPaddingOnTensors()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 437 Value outputTensor, in makeTransposeOp() argument
439 auto resultTensorType = outputTensor.getType().cast<RankedTensorType>(); in makeTransposeOp()
460 loc, resultTensorType, inputTensor, outputTensor, in makeTransposeOp()
980 Value outputTensor = operands[opOperand->getOperandNumber()]; in insertSlicesBack() local
981 if (auto sliceOp = outputTensor.getDefiningOp<tensor::ExtractSliceOp>()) { in insertSlicesBack()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Utils/ |
| H A D | Utils.h | 138 Value outputTensor,
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Shape/IR/ |
| H A D | Shape.cpp | 1756 TensorType outputTensor = outputs[0].dyn_cast<TensorType>(); in areCastCompatible() local
1757 return outputTensor && outputTensor.getElementType().isa<IndexType>(); in areCastCompatible()
|