Searched refs:inputTensor (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Fusion.cpp | 405 Value inputTensor = consumerOpOperand.get(); in fuseProducerOfTensor() local
407 getProducerOfTensor(inputTensor, producerOpResult); in fuseProducerOfTensor()
426 Value inputTensor = consumerOpOperand.get(); in fuseProducerOfTensor() local
429 auto sliceOp = inputTensor.getDefiningOp<tensor::ExtractSliceOp>(); in fuseProducerOfTensor()
432 << "\nNot fusable, not an extract_slice op: " << inputTensor); in fuseProducerOfTensor()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Utils/ |
| H A D | Utils.h | 137 GenericOp makeTransposeOp(OpBuilder &b, Location loc, Value inputTensor,
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Shape/IR/ |
| H A D | Shape.cpp | 1748 if (auto inputTensor = inputs[0].dyn_cast<RankedTensorType>()) { in areCastCompatible() local
1749 if (!inputTensor.getElementType().isa<IndexType>() || in areCastCompatible()
1750 inputTensor.getRank() != 1) in areCastCompatible()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 436 GenericOp makeTransposeOp(OpBuilder &b, Location loc, Value inputTensor, in makeTransposeOp() argument
460 loc, resultTensorType, inputTensor, outputTensor, in makeTransposeOp()
|