Home
last modified time | relevance | path

Searched defs:sizeN (Results 1 – 4 of 4) sorted by relevance

/llvm-project/mlir/lib/Dialect/NVGPU/IR/
H A DNVGPUDialect.cpp530 isAllowedSizeN(int sizeN,Type typeA) isAllowedSizeN() argument
634 int64_t sizeN = accType.getFragmented().getDimSize(1); verify() local
/llvm-project/mlir/lib/Conversion/NVGPUToNVVM/
H A DNVGPUToNVVM.cpp432 int64_t sizeN = type.getFragmented().getDimSize(1); runOnOperation() local
1061 int64_t sizeN = op.getTensorMap().getType().getTensor().getDimSize(0); matchAndRewrite() local
1251 findWgmmaShape(int64_t sizeM,int64_t sizeN,Type inputElemType) findWgmmaShape() argument
/llvm-project/mlir/lib/Dialect/LLVMIR/IR/
H A DNVVMDialect.cpp798 isAllowedSizeN(int sizeN,NVVM::WGMMATypes typeA) isAllowedSizeN() argument
/llvm-project/mlir/lib/Dialect/NVGPU/TransformOps/
H A DNVGPUTransformOps.cpp83 int64_t sizeN = type.getFragmented().getDimSize(1); in populatePatterns() local