Searched refs:matrixType (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/SPIRV/IR/ |
| H A D | SPIRVTypes.cpp | 125 if (auto matrixType = dyn_cast<MatrixType>()) in getNumElements() local
126 return matrixType.getNumColumns(); in getNumElements()
676 } else if (auto matrixType = dyn_cast<MatrixType>()) { in getExtensions() local
677 matrixType.getExtensions(extensions, storage); in getExtensions()
696 } else if (auto matrixType = dyn_cast<MatrixType>()) { in getCapabilities() local
697 matrixType.getCapabilities(capabilities, storage); in getCapabilities()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/GPUToNVVM/ |
| H A D | WmmaOpsToNvvm.cpp | 160 auto matrixType = adaptor.src().getType().cast<LLVM::LLVMStructType>(); in matchAndRewrite() local
161 for (unsigned i = 0, e = matrixType.getBody().size(); i < e; ++i) { in matchAndRewrite()
163 loc, matrixType.getBody()[i], adaptor.src(), in matchAndRewrite()
|
| /llvm-project-15.0.7/clang/lib/AST/ |
| H A D | Type.cpp | 285 MatrixType::MatrixType(TypeClass tc, QualType matrixType, QualType canonType, in MatrixType() argument
288 (RowExpr ? (matrixType->getDependence() | TypeDependence::Dependent | in MatrixType()
290 (matrixType->isVariablyModifiedType() in MatrixType()
293 (matrixType->containsUnexpandedParameterPack() || in MatrixType()
300 : matrixType->getDependence())), in MatrixType()
301 ElementType(matrixType) {} in MatrixType()
303 ConstantMatrixType::ConstantMatrixType(QualType matrixType, unsigned nRows, in ConstantMatrixType() argument
305 : ConstantMatrixType(ConstantMatrix, matrixType, nRows, nColumns, in ConstantMatrixType()
308 ConstantMatrixType::ConstantMatrixType(TypeClass tc, QualType matrixType, in ConstantMatrixType() argument
311 : MatrixType(tc, matrixType, canonType), NumRows(nRows), in ConstantMatrixType()
|
| /llvm-project-15.0.7/mlir/lib/Target/SPIRV/Serialization/ |
| H A D | Serializer.cpp | 596 if (auto matrixType = type.dyn_cast<spirv::MatrixType>()) { in prepareBasicType() local
598 if (failed(processTypeImpl(loc, matrixType.getColumnType(), elementTypeID, in prepareBasicType()
604 operands.push_back(matrixType.getNumColumns()); in prepareBasicType()
|