Searched defs:tensorMap (Results 1 – 2 of 2) sorted by relevance
| /llvm-project/mlir/lib/ExecutionEngine/ |
| H A D | CudaRuntimeWrappers.cpp | 393 mgpuTensorMapEncodeTiled(CUtensorMap * tensorMap,CUtensorMapDataType tensorDataType,cuuint32_t tensorRank,void * globalAddress,const cuuint64_t * globalDim,const cuuint64_t * globalStrides,const cuuint32_t * boxDim,const cuuint32_t * elementStrides,CUtensorMapInterleave interleave,CUtensorMapSwizzle swizzle,CUtensorMapL2promotion l2Promotion,CUtensorMapFloatOOBfill oobFill) mgpuTensorMapEncodeTiled() argument
459 CUtensorMap tensorMap; mgpuTensorMapEncodeTiledMemref() local
|
| /llvm-project/mlir/lib/Conversion/NVGPUToNVVM/ |
| H A D | NVGPUToNVVM.cpp | 1202 Value tensorMap = matchAndRewrite() local
|