Home
last modified time | relevance | path

Searched refs:outputTensor (Results 1 – 6 of 6) sorted by relevance

/llvm-project/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp735 Value outputTensor = operands[opOperand.getOperandNumber()]; in computeTileSizes()
736 if (auto sliceOp = outputTensor.getDefiningOp<tensor::ExtractSliceOp>()) { in computeTileSizes()
253 makeTransposeOp(OpBuilder & b,Location loc,Value inputTensor,Value outputTensor,ArrayRef<int64_t> transposeVector) makeTransposeOp() argument
761 Value outputTensor = operands[opOperand.getOperandNumber()]; insertSlicesBack() local
/llvm-project/mlir/include/mlir/Dialect/Linalg/Utils/
H A DUtils.h
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DConvertConv2DToImg2Col.cpp248 Value outputTensor = rewriter.create<tensor::EmptyOp>( in rewriteInIm2Col() local
260 loc, outputTensor.getType(), in rewriteInIm2Col()
261 /*inputs=*/operand, /*outputs=*/outputTensor, indexingMaps, in rewriteInIm2Col()
H A DHoistPadding.cpp639 Value outputTensor = rewriter.create<tensor::ExtractSliceOp>( in buildPackingLoopNestImpl() local
643 loc, paddedTensor, outputTensor, transposeVector); in buildPackingLoopNestImpl()
/llvm-project/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalg.cpp810 Value outputTensor = opBuilder.create<tensor::EmptyOp>( in broadcastDynamicDimension()
817 loc, outputTensor.getType(), operand, outputTensor, affineMaps, in broadcastDynamicDimension()
885 Value outputTensor = rewriter.create<tensor::EmptyOp>( in emitElementwiseComputation()
913 loc, outputTensor.getType(), operands, outputTensor, affineMaps, in elementwiseMatchAndRewriteHelper()
788 Value outputTensor = opBuilder.create<tensor::EmptyOp>( broadcastDynamicDimension() local
863 Value outputTensor = rewriter.create<tensor::EmptyOp>( emitElementwiseComputation() local
/llvm-project/mlir/lib/Dialect/Shape/IR/
H A DShape.cpp1923 TensorType outputTensor = llvm::dyn_cast<TensorType>(outputs[0]); in build()
1924 return outputTensor && llvm::isa<IndexType>(outputTensor.getElementType()); in build()
1897 TensorType outputTensor = llvm::dyn_cast<TensorType>(outputs[0]); areCastCompatible() local