Home
last modified time | relevance | path

Searched defs:weightType (Results 1 – 6 of 6) sorted by relevance

/llvm-project/mlir/lib/Dialect/Tosa/Utils/
H A DQuantUtils.cpp124 auto weightType = dyn_cast<ShapedType>(weight.getType()); in buildConvOpQuantizationAttr() local
243 auto weightType = dyn_cast<ShapedType>(weight.getType()); in buildConvOpResultTypeInfo() local
/llvm-project/mlir/test/lib/Dialect/Tosa/
H A DTosaTestPasses.cpp121 auto weightType = in matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaDecomposeConv2D.cpp38 ShapedType weightType = cast<ShapedType>(weight.getType()); in matchAndRewrite() local
H A DTosaDecomposeDepthwise.cpp33 ShapedType weightType = cast<ShapedType>(weight.getType()); matchAndRewrite() local
H A DTosaValidation.cpp303 if (ShapedType weightType = in levelCheckConv() local
/llvm-project/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp227 weightType = llvm::dyn_cast<RankedTensorType>(op.getWeight().getType()); in verifyConvOp() local