Home
last modified time | relevance | path

Searched defs:outputType (Results 1 – 19 of 19) sorted by relevance

/llvm-project/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaMakeBroadcastable.cpp41 RankedTensorType outputType, Value &input1, in reshapeLowerToHigher() argument
90 auto outputType = dyn_cast<RankedTensorType>(output.getType()); in matchAndRewrite() local
118 auto outputType = dyn_cast<RankedTensorType>(output.getType()); in matchAndRewrite() local
149 auto outputType = dyn_cast<RankedTensorType>(output.getType()); matchAndRewrite() local
177 auto outputType = dyn_cast<RankedTensorType>(output.getType()); matchAndRewrite() local
[all...]
H A DTosaFolders.cpp136 ShapedType outputType, in transposeType()
184 ShapedType outputType, in transpose()
215 auto outputType = cast<ShapedType>(op.getType()); in matchAndRewrite() local
/llvm-project/mlir/utils/vscode/src/PDLL/commands/
H A DviewOutput.ts38 const outputType = constant
/llvm-project/mlir/test/lib/Dialect/Tosa/
H A DTosaTestPasses.cpp51 auto outputType = in matchAndRewrite() local
129 auto outputType = in matchAndRewrite() local
/llvm-project/mlir/tools/mlir-pdll/
H A Dmlir-pdll.cpp40 OutputType outputType, std::vector<std::string> &includeDirs, in processBuffer() argument
153 llvm::cl::opt<enum OutputType> outputType( in main() local
/llvm-project/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp366 buildConvOpWithQuantInfo(OpBuilder & builder,OperationState & result,Type outputType,Value input,Value weight,Value bias,DenseI64ArrayAttr pad,DenseI64ArrayAttr stride,DenseI64ArrayAttr dilation) buildConvOpWithQuantInfo() argument
388 buildTransConvOpWithQuantInfo(OpBuilder & builder,OperationState & result,Type outputType,Value input,Value weight,Value bias,DenseI64ArrayAttr outpad,DenseI64ArrayAttr stride,DenseI64ArrayAttr outputShape) buildTransConvOpWithQuantInfo() argument
409 buildFCOpWithQuantInfo(OpBuilder & builder,OperationState & result,Type outputType,Value input,Value weight,Value bias) buildFCOpWithQuantInfo() argument
428 buildMatMulOpWithQuantInfo(OpBuilder & builder,OperationState & result,Type outputType,Value a,Value b) buildMatMulOpWithQuantInfo() argument
465 buildAvgPool2dOpWithQuantInfo(OpBuilder & builder,OperationState & result,Type outputType,Value input,DenseArrayAttr kernel,DenseArrayAttr stride,DenseArrayAttr pad,TypeAttr accType) buildAvgPool2dOpWithQuantInfo() argument
483 buildUnaryOpWithQuantInfo(OpBuilder & builder,OperationState & result,Type outputType,Value input) buildUnaryOpWithQuantInfo() argument
496 buildPadOpWithQuantInfo(OpBuilder & builder,OperationState & result,Type outputType,Value input,Value paddings) buildPadOpWithQuantInfo() argument
509 buildExplicitValuePadOpWithQuantInfo(OpBuilder & builder,OperationState & result,Type outputType,Value input,Value paddings,Value padConst) buildExplicitValuePadOpWithQuantInfo() argument
896 ShapedType outputType = llvm::cast<ShapedType>(getType()); verify() local
960 RankedTensorType outputType = getType(); verify() local
1086 TensorType outputType = getOutput().getType(); verify() local
1298 TensorType outputType = op.getOutput().getType(); REDUCE_SHAPE_INFER() local
1895 TensorType outputType = getOutput().getType(); verify() local
[all...]
H A DTosaCanonicalizations.cpp248 ShapedType outputType = llvm::cast<ShapedType>(output.getType()); in matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Tosa/Utils/
H A DQuantUtils.cpp197 auto outputType = dyn_cast<ShapedType>(outputRawType); in buildUnaryOpQuantizationAttr() local
239 Type mlir::tosa::buildConvOpResultTypeInfo(OpBuilder &builder, Type outputType, in buildConvOpResultTypeInfo()
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DConvertConv2DToImg2Col.cpp82 auto outputType = cast<ShapedType>(convOp.getOutputs()[0].getType()); in rewriteInIm2Col() local
218 auto outputType = cast<RankedTensorType>(convOp.getOutputs()[0].getType()); in rewriteInIm2Col() local
368 auto outputType = cast<ShapedType>(convOp.getOutputs()[0].getType()); in rewriteInIm2Col() local
501 auto outputType = cast<ShapedType>(convOp.getOutputs()[0].getType()); in rewriteInIm2Col() local
H A DConstantFold.cpp68 auto outputType = dyn_cast<ShapedType>(linalgOp->getResultTypes().front()); in matchAndRewrite() local
H A DTransforms.cpp1331 auto outputType = dyn_cast<RankedTensorType>(output.getType()); returningMatchAndRewrite() local
1455 auto outputType = dyn_cast<RankedTensorType>(output.getType()); returningMatchAndRewrite() local
1523 auto outputType = dyn_cast<RankedTensorType>(output.getType()); returningMatchAndRewrite() local
H A DVectorization.cpp1139 auto outputType = dyn_cast<VectorType>(outputVec.getType()); reduceIfNeeded() local
/llvm-project/mlir/lib/Transforms/Utils/
H A DDialectConversion.cpp1395 Type outputType = origOutputType; applySignatureConversion() local
1427 buildUnresolvedMaterialization(MaterializationKind kind,Block * insertBlock,Block::iterator insertPt,Location loc,ValueRange inputs,Type outputType,Type origOutputType,const TypeConverter * converter) buildUnresolvedMaterialization() argument
1444 buildUnresolvedArgumentMaterialization(Block * block,Location loc,ValueRange inputs,Type origOutputType,Type outputType,const TypeConverter * converter) buildUnresolvedArgumentMaterialization() argument
1450 buildUnresolvedTargetMaterialization(Location loc,Value input,Type outputType,const TypeConverter * converter) buildUnresolvedTargetMaterialization() argument
2724 Type outputType = opResult.getType(); computeNecessaryMaterializations() local
2807 Type outputType = opResult.getType(); legalizeUnresolvedMaterialization() local
/llvm-project/lld/MachO/
H A DConfig.h198 llvm::MachO::HeaderFileType outputType; global() member
/llvm-project/mlir/lib/Dialect/SparseTensor/IR/
H A DSparseTensorDialect.cpp1732 verifyNumBlockArgs(T * op,Region & region,const char * regionName,TypeRange inputTypes,Type outputType) verifyNumBlockArgs() argument
1761 Type outputType = getOutput().getType(); verify() local
1796 Type outputType = getOutput().getType(); verify() local
[all...]
/llvm-project/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalg.cpp2354 auto outputType = computeOutputShape(rewriter, loc, input, dynamicSizes); matchAndRewrite() local
2471 auto outputType = RankedTensorType::get(staticSizes, real_el_ty); matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp1262 for (Type outputType : outputTypes) { parseDstStyleOp() local
2448 ShapedType outputType = getOutputOperandType(); verify() local
[all...]
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseTensorRewriting.cpp334 auto outputType = getRankedTensorType(op.getResult(0)); in matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/LLVMIR/IR/
H A DLLVMDialect.cpp2812 IntegerType inputType, outputType; verifyExtOp() local