Home
last modified time | relevance | path

Searched defs:inputType (Results 1 – 25 of 37) sorted by relevance

12

/llvm-project/mlir/lib/Dialect/Tosa/Utils/
H A DQuantUtils.cpp110 #define GET_UQTYPE(inputType) \ argument
112 #define GET_QTYPE(inputType) \ argument
123 auto inputType = dyn_cast<ShapedType>(input.getType()); in buildConvOpQuantizationAttr() local
196 auto inputType = dyn_cast<ShapedType>(input.getType()); in buildUnaryOpQuantizationAttr() local
222 auto inputType = dyn_cast<ShapedType>(input.getType()); in buildPadOpQuantizationAttr() local
242 auto inputType = dyn_cast<ShapedType>(input.getType()); in buildConvOpResultTypeInfo() local
/llvm-project/mlir/lib/Dialect/Quant/Utils/
H A DUniformSupport.cpp16 static bool isQuantizablePrimitiveType(Type inputType) { in isQuantizablePrimitiveType() argument
21 ExpressedToQuantizedConverter::forInputType(Type inputType) { in forInputType() argument
[all...]
/llvm-project/mlir/test/lib/Dialect/Tosa/
H A DTosaTestPasses.cpp44 auto inputType = in matchAndRewrite() local
114 auto inputType = in matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp226 auto inputType = llvm::dyn_cast<RankedTensorType>(op.getInput().getType()); verifyConvOp() local
276 const auto inputType = llvm::cast<ShapedType>(getInput().getType()); verify() local
285 auto inputType = llvm::cast<ShapedType>(getInput().getType()); verify() local
436 auto inputType = llvm::dyn_cast<ShapedType>(a.getType()); buildMatMulOpWithQuantInfo() local
656 Type inputType = inferReturnTypeComponents() local
836 auto inputType = llvm::dyn_cast<RankedTensorType>(getInput().getType()); verify() local
895 ShapedType inputType = llvm::cast<ShapedType>(getInput1().getType()); verify() local
924 Type inputType = getElementTypeOrSelf(adaptor.getInput1().getType()); inferReturnTypeComponents() local
959 TensorType inputType = getInput1().getType(); verify() local
1084 TensorType inputType = getInput1().getType(); verify() local
1131 auto inputType = cast<TensorType>(input.getType()); reifyResultShapes() local
1246 ReduceInferReturnTypes(ShapeAdaptor operandShape,Type inputType,IntegerAttr axis,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) ReduceInferReturnTypes() argument
1297 TensorType inputType = op.getInput().getType(); REDUCE_SHAPE_INFER() local
1894 TensorType inputType = getInput().getType(); verify() local
[all...]
H A DTosaCanonicalizations.cpp247 ShapedType inputType = llvm::cast<ShapedType>(input.getType()); in matchAndRewrite() local
281 auto inputType = llvm::dyn_cast<RankedTensorType>(op.getInput().getType()); in matchAndRewrite() local
404 auto inputType = dyn_cast<RankedTensorType>(input.getType()); matchAndRewrite() local
[all...]
/llvm-project/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaFolders.cpp135 DenseElementsAttr transposeType(const RangeType &data, ShapedType inputType, in transposeType()
183 DenseElementsAttr transpose(ElementsAttr attr, ShapedType inputType, in transpose()
235 auto inputType = cast<ShapedType>(op.getInput1().getType()); in matchAndRewrite() local
H A DTosaDecomposeConv2D.cpp37 ShapedType inputType = cast<ShapedType>(input.getType()); in matchAndRewrite() local
H A DTosaDecomposeDepthwise.cpp32 ShapedType inputType = cast<ShapedType>(input.getType()); matchAndRewrite() local
[all...]
/llvm-project/mlir/examples/toy/Ch3/mlir/
H A DDialect.cpp279 auto inputType = *operand_type_begin(); in verify() local
303 auto inputType = llvm::dyn_cast<RankedTensorType>(getOperand().getType()); in verify() local
/llvm-project/mlir/examples/toy/Ch2/mlir/
H A DDialect.cpp279 auto inputType = *operand_type_begin(); in verify() local
303 auto inputType = llvm::dyn_cast<RankedTensorType>(getOperand().getType()); in verify() local
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DTransposeConv2D.cpp68 auto inputType = op->getOperand(0).getType(); in transposeConv2DHelper() local
H A DConvertConv2DToImg2Col.cpp80 auto inputType = cast<ShapedType>(convOp.getInputs()[0].getType()); in rewriteInIm2Col() local
216 auto inputType = cast<RankedTensorType>(convOp.getInputs()[0].getType()); in rewriteInIm2Col() local
366 auto inputType = cast<ShapedType>(convOp.getInputs()[0].getType()); in rewriteInIm2Col() local
499 auto inputType = cast<ShapedType>(convOp.getInputs()[0].getType()); in rewriteInIm2Col() local
H A DTransforms.cpp1037 ShapedType inputType = packOp.getSourceType(); getPackOpSourceOrPaddedSource() local
1329 auto inputType = dyn_cast<RankedTensorType>(input.getType()); returningMatchAndRewrite() local
1453 auto inputType = dyn_cast<RankedTensorType>(input.getType()); returningMatchAndRewrite() local
1521 auto inputType = dyn_cast<RankedTensorType>(input.getType()); returningMatchAndRewrite() local
[all...]
/llvm-project/mlir/lib/Dialect/ArmSME/Transforms/
H A DOuterProductFusion.cpp58 isCompatible(PatternRewriter & rewriter,arm_sme::OuterProductOp op,VectorType resultType,VectorType inputType) isCompatible() argument
86 auto inputType = cast<VectorType>(lhs.getType()); createInterleave2Intrinsic() local
435 __anon67516db70e02(VectorType resultType, VectorType inputType, auto lhsExtendOp, auto rhsExtendOp) canFuseOuterProducts() argument
/llvm-project/mlir/examples/toy/Ch4/mlir/
H A DDialect.cpp394 auto inputType = *operand_type_begin(); in verify() local
424 auto inputType = llvm::dyn_cast<RankedTensorType>(getOperand().getType()); in verify() local
/llvm-project/mlir/examples/toy/Ch6/mlir/
H A DDialect.cpp394 auto inputType = *operand_type_begin(); in verify() local
424 auto inputType = llvm::dyn_cast<RankedTensorType>(getOperand().getType()); in verify() local
/llvm-project/mlir/examples/toy/Ch5/mlir/
H A DDialect.cpp394 auto inputType = *operand_type_begin(); in verify() local
424 auto inputType = llvm::dyn_cast<RankedTensorType>(getOperand().getType()); in verify() local
/llvm-project/mlir/lib/Conversion/TosaToTensor/
H A DTosaToTensor.cpp48 TensorType inferReshapeExpandedType(TensorType inputType, in inferReshapeExpandedType() argument
241 auto inputType = inferReshapeInputType(input, newShape); in matchAndRewrite() local
/llvm-project/mlir/include/mlir/Dialect/Quant/
H A DUniformSupport.h
/llvm-project/mlir/lib/Conversion/AMDGPUToROCDL/
H A DAMDGPUToROCDL.cpp336 Type inputType = input.getType(); mfmaConcatIfNeeded() local
373 Type inputType = llvmInput.getType(); wmmaPushInputOperand() local
416 Type inputType = output.getType(); wmmaPushOutputOperand() local
/llvm-project/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransferOpTransforms.cpp278 MemRefType inputType = cast<MemRefType>(input.getType()); in dropUnitDims() argument
293 MemRefType inputType = cast<MemRefType>(input.getType()); rankReducingSubviewDroppingUnitDims() local
495 ShapedType inputType = cast<ShapedType>(input.getType()); collapseInnerDims() local
[all...]
/llvm-project/mlir/test/lib/Conversion/OneToNTypeConversion/
H A DTestOneToNTypeConversionPass.cpp152 TupleType inputType = dyn_cast<TupleType>(input.getType()); buildGetTupleElementOps() local
/llvm-project/mlir/examples/toy/Ch7/mlir/
H A DDialect.cpp428 auto inputType = *operand_type_begin(); in verify() local
486 auto inputType = llvm::dyn_cast<RankedTensorType>(getOperand().getType()); in verify() local
/llvm-project/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp1672 auto inputType = llvm::cast<ShapedType>(getInputs()[0].getType()); verify() local
1809 auto inputType = getInput().getType(); verify() local
1962 auto inputType = getInput().getType(); verify() local
2447 ShapedType inputType = getInputOperandType(); verify() local
2615 auto inputType = cast<ShapedType>(input.getType()); reduce() local
2638 auto inputType = cast<ShapedType>(input.getType()); buildSubAndExpOp() local
2664 auto inputType = cast<ShapedType>(numerator.getType()); buildDivOp() local
2708 ShapedType inputType = getInputOperandType(); decomposeOperation() local
[all...]
/llvm-project/mlir/lib/Interfaces/
H A DControlFlowInterfaces.cpp133 Type inputType = std::get<1>(typesIdx.value()); in verifyTypesAlongAllEdges() local

12