Searched defs:useNvGpu (Results 1 – 1 of 1) sorted by relevance
78 bool useNvGpu) { in contractSupportsMMAMatrixType() argument 275 supportsMMaMatrixType(Operation * op,bool useNvGpu) supportsMMaMatrixType() argument 344 getOpToConvert(mlir::Operation * op,bool useNvGpu) getOpToConvert() argument 1230 populatePrepareVectorToMMAPatterns(RewritePatternSet & patterns,bool useNvGpu) populatePrepareVectorToMMAPatterns() argument 1344 createConvertVectorToGPUPass(bool useNvGpu) createConvertVectorToGPUPass() argument [all...]